Home
last modified time | relevance | path

Searched refs:saved_model_builder (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/saved_model/
Dsaved_model_test.py42 from tensorflow.python.saved_model import builder as saved_model_builder unknown
225 builder = saved_model_builder._SavedModelBuilder(export_dir)
247 builder = saved_model_builder._SavedModelBuilder(export_dir)
266 builder = saved_model_builder._SavedModelBuilder(export_dir)
364 builder = saved_model_builder._SavedModelBuilder(export_dir)
417 builder = saved_model_builder._SavedModelBuilder(export_dir)
453 builder = saved_model_builder._SavedModelBuilder(export_dir)
472 self.assertRaises(AssertionError, saved_model_builder._SavedModelBuilder,
477 builder = saved_model_builder._SavedModelBuilder(export_dir)
511 builder = saved_model_builder._SavedModelBuilder(export_dir)
[all …]
Dloader_test.py34 from tensorflow.python.saved_model import builder as saved_model_builder unknown
72 @parameterized.parameters((saved_model_builder.SavedModelBuilder,),
73 (saved_model_builder._SavedModelBuilder,))
92 if builder_cls == saved_model_builder._SavedModelBuilder:
159 if builder_cls == saved_model_builder._SavedModelBuilder:
262 builder = saved_model_builder.SavedModelBuilder(path)
/external/tensorflow/tensorflow/python/tools/
Dsaved_model_utils_test.py28 from tensorflow.python.saved_model import builder as saved_model_builder unknown
47 builder = saved_model_builder.SavedModelBuilder(saved_model_dir)
69 builder = saved_model_builder.SavedModelBuilder(saved_model_dir)
Dfreeze_graph_test.py43 from tensorflow.python.saved_model import builder as saved_model_builder unknown
153 builder = saved_model_builder.SavedModelBuilder(path)
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/testdata/
Dgen_tftrt_model.py117 saved_model_builder = builder.SavedModelBuilder(tf_saved_model_dir)
120 saved_model_builder.add_meta_graph_and_variables(
125 saved_model_builder.save()
/external/tensorflow/tensorflow/python/keras/saving/
Dsaved_model_experimental.py37 from tensorflow.python.saved_model import builder as saved_model_builder unknown
186 builder = saved_model_builder._SavedModelBuilder(path) # pylint: disable=protected-access
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py770 saved_model_builder = builder.SavedModelBuilder(output_saved_model_dir)
779 saved_model_builder.add_meta_graph_and_variables(
784 saved_model_builder.save()
Dtrt_convert_test.py168 saved_model_builder = builder.SavedModelBuilder(input_saved_model_dir)
171 saved_model_builder.add_meta_graph_and_variables(
174 saved_model_builder.save()
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dtf_trt_integration_test_base.py944 saved_model_builder = builder.SavedModelBuilder(saved_model_dir)
948 saved_model_builder.add_meta_graph_and_variables(
954 saved_model_builder.save()