code: def test_split_saved_model(self):
      sizes = [100, 100, 1000, 100, 1000, 500, 100, 100, 100]
      fn1 = [100, 100, 100]
      fn2 = [100, 500]
      fn3 = [100]
      fn4 = [100, 100]
  
      max_size = 500
      constants.debug_set_max_size(max_size)
  
      graph_def = test_util.make_graph_def_with_constant_nodes(
          sizes, fn1=fn1, fn2=fn2, fn3=fn3, fn4=fn4
      )
      proto = saved_model_pb2.SavedModel()
      proto.meta_graphs.add().graph_def.CopyFrom(graph_def)
  
      splitter = saved_model.SavedModelSplitter(proto)
      chunks, _ = splitter.split()
      self._assert_chunk_sizes(chunks, max_size)
  
file_path: tensorflow-upstream/tensorflow/tools/proto_splitter/python/saved_model_test.py
imports: from google.protobuf import message
  from tensorflow.core.protobuf import saved_model_pb2
  from tensorflow.python.platform import test
  from tensorflow.tools.proto_splitter import constants
  from tensorflow.tools.proto_splitter.python import saved_model
  from tensorflow.tools.proto_splitter.python import test_util
label_desc: Use image node_id 2 for calling the SavedModelSplitterTest obj's underlying member method code with example usage: obj.test_split_saved_model() without return types