code: def test_merge_graph_defs_partitioned_call_remap(self):
      expected_proto = GraphDef()
      text_format.Parse(
          """
                  node {
                    name: "graph_1/X"
                    op: "PartitionedCall"
                    attr {
                      key: "f"
                      value {
                        func {
                          name: "graph_1_foo"
                        }
                      }
                    }
                  }
                  library {
                    function {
                      signature {
                        name: "graph_1_foo"
                        input_arg {
                          name: "x"
                          type: DT_HALF
                        }
                        output_arg {
                          name: "identity"
                          type: DT_HALF
                        }
                      }
                    }
                  }
              """,
          expected_proto,
      )
  
      graph_def_a = GraphDef()
      text_format.Parse(
          """
                  node {
                    name: "X"
                    op: "PartitionedCall"
                    attr {
                      key: "f"
                      value {
                        func {
                          name: "foo"
                        }
                      }
                    }
                  }
                  library {
                    function {
                      signature {
                        name: "foo"
                        input_arg {
                          name: "x"
                          type: DT_HALF
                        }
                        output_arg {
                          name: "identity"
                          type: DT_HALF
                        }
                      }
                    }
                  }
              """,
          graph_def_a,
      )
      graph_def_b = GraphDef()
  
      self.assertProtoEquals(
          expected_proto,
          graph_util.merge_graph_defs([graph_def_a, graph_def_b]),
      )
  
file_path: tensorboard/tensorboard/plugins/graph/graph_util_test.py
imports: from google.protobuf import text_format
  import tensorflow
  from tensorboard.compat.proto.graph_pb2 import GraphDef
  from tensorboard.plugins.graph import graph_util
label_desc: Use image node_id 4 for calling the GraphUtilTest obj's underlying member method code with example usage: obj.test_merge_graph_defs_partitioned_call_remap() without return types