code: def setUp(self):
      super(GraphInterfaceTest, self).setUp()
      self._cost_graph = cost_graph_pb2.CostGraphDef(
          node=[
              cost_graph_pb2.CostGraphDef.Node(
                  name="X",
                  device="/device:CPU:0",
                  id=0,
                  output_info=[
                      cost_graph_pb2.CostGraphDef.Node.OutputInfo(
                          size=48,
                          alias_input_port=-1,
                          dtype=types_pb2.DT_INT32,
                          shape=tensor_shape_pb2.TensorShapeProto(
                              dim=[
                                  tensor_shape_pb2.TensorShapeProto.Dim(
                                      size=3
                                  ),
                                  tensor_shape_pb2.TensorShapeProto.Dim(
                                      size=4
                                  ),
                              ]
                          ),
                      ),
                  ],
              ),
              cost_graph_pb2.CostGraphDef.Node(
                  name="Y",
                  device="/device:CPU:0",
                  id=1,
                  output_info=[
                      cost_graph_pb2.CostGraphDef.Node.OutputInfo(
                          size=80,
                          alias_input_port=-1,
                          dtype=types_pb2.DT_INT32,
                          shape=tensor_shape_pb2.TensorShapeProto(
                              dim=[
                                  tensor_shape_pb2.TensorShapeProto.Dim(
                                      size=4
                                  ),
                                  tensor_shape_pb2.TensorShapeProto.Dim(
                                      size=5
                                  ),
                              ]
                          ),
                      ),
                  ],
              ),
              cost_graph_pb2.CostGraphDef.Node(
                  name="Z1",
                  device="/device:CPU:0",
                  id=2,
                  input_info=[
                      cost_graph_pb2.CostGraphDef.Node.InputInfo(
                          preceding_node=0,
                          preceding_port=0,
                      ),
                      cost_graph_pb2.CostGraphDef.Node.InputInfo(
                          preceding_node=1,
                          preceding_port=0,
                      ),
                  ],
                  output_info=[
                      cost_graph_pb2.CostGraphDef.Node.OutputInfo(
                          size=60,
                          alias_input_port=-1,
                          dtype=types_pb2.DT_INT32,
                          shape=tensor_shape_pb2.TensorShapeProto(
                              dim=[
                                  tensor_shape_pb2.TensorShapeProto.Dim(
                                      size=3
                                  ),
                                  tensor_shape_pb2.TensorShapeProto.Dim(
                                      size=5
                                  ),
                              ]
                          ),
                      ),
                  ],
                  is_final=True,
              ),
              cost_graph_pb2.CostGraphDef.Node(
                  name="Z2",
                  device="/device:CPU:0",
                  id=3,
                  input_info=[
                      cost_graph_pb2.CostGraphDef.Node.InputInfo(
                          preceding_node=0,
                          preceding_port=0,
                      ),
                      cost_graph_pb2.CostGraphDef.Node.InputInfo(
                          preceding_node=1,
                          preceding_port=0,
                      ),
                  ],
                  output_info=[
                      cost_graph_pb2.CostGraphDef.Node.OutputInfo(
                          size=60,
                          alias_input_port=-1,
                          dtype=types_pb2.DT_INT32,
                          shape=tensor_shape_pb2.TensorShapeProto(
                              dim=[
                                  tensor_shape_pb2.TensorShapeProto.Dim(
                                      size=3
                                  ),
                                  tensor_shape_pb2.TensorShapeProto.Dim(
                                      size=5
                                  ),
                              ]
                          ),
                      ),
                  ],
              ),
          ]
      )
      self._sizeless_cost_graph = self.StripCostGraphDef(
          self._cost_graph, "SIZES"
      )
      self._deviceless_cost_graph = self.StripCostGraphDef(
          self._cost_graph, "DEVICES"
      )
  
      self._cost_graph_string = self._cost_graph.SerializeToString()
      self._sizeless_cost_graph_string = (
          self._sizeless_cost_graph.SerializeToString()
      )
      self._deviceless_cost_graph_string = (
          self._deviceless_cost_graph.SerializeToString()
      )
  
file_path: mesh/mesh_tensorflow/auto_mtf/graph_interface_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import mesh_tensorflow
  from mesh_tensorflow.auto_mtf import graph_interface
  import tensorflow.compat.v1
  from tensorflow.core.framework import cost_graph_pb2
  from tensorflow.core.framework import tensor_shape_pb2
  from tensorflow.core.framework import types_pb2
label_desc: Use image node_id 1 for calling the GraphInterfaceTest obj's underlying member method code with example usage: obj.setUp() without return types