Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/cc/tools/
Dfreeze_saved_model_test.cc102 const GraphDef& frozen_graph_def, in RunAndCompareFrozenAndUnfrozenGraphs() argument
110 TF_ASSERT_OK(frozen_session->Create(frozen_graph_def)); in RunAndCompareFrozenAndUnfrozenGraphs()
145 GraphDef frozen_graph_def; in TestFreezeGraphWithoutDependentVariables() local
148 TF_ASSERT_OK(FreezeSavedModel(saved_model_bundle, &frozen_graph_def, in TestFreezeGraphWithoutDependentVariables()
159 GraphDefEqual(frozen_graph_def, expected_graph_def); in TestFreezeGraphWithoutDependentVariables()
162 frozen_graph_def, "c:0"); in TestFreezeGraphWithoutDependentVariables()
190 GraphDef frozen_graph_def; in TestFreezeGraphWithDependentVariables() local
193 TF_ASSERT_OK(FreezeSavedModel(saved_model_bundle, &frozen_graph_def, in TestFreezeGraphWithDependentVariables()
201 EXPECT_EQ(frozen_graph_def.node_size(), expected_nodes); in TestFreezeGraphWithDependentVariables()
202 for (const NodeDef& node : frozen_graph_def.node()) { in TestFreezeGraphWithDependentVariables()
[all …]
Dfreeze_saved_model.cc177 GraphDef* frozen_graph_def) { in FreezeGraphDef() argument
180 *frozen_graph_def->mutable_versions() = graph_def.versions(); in FreezeGraphDef()
181 *frozen_graph_def->mutable_library() = graph_def.library(); in FreezeGraphDef()
206 frozen_graph_def->add_node()); in FreezeGraphDef()
212 ConvertReadVariableOpToIdentity(node, frozen_graph_def->add_node()); in FreezeGraphDef()
215 *frozen_graph_def->add_node() = node; in FreezeGraphDef()
224 GraphDef* frozen_graph_def, in FreezeSavedModel() argument
229 FreezeGraphDef(saved_model_bundle, *outputs, frozen_graph_def)); in FreezeSavedModel()
Dfreeze_saved_model.h37 GraphDef* frozen_graph_def,
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfreeze_requantization_ranges.cc170 GraphDef frozen_graph_def; in FreezeRequantizationRanges() local
179 NodeDef* min_node = frozen_graph_def.mutable_node()->Add(); in FreezeRequantizationRanges()
189 NodeDef* max_node = frozen_graph_def.mutable_node()->Add(); in FreezeRequantizationRanges()
198 NodeDef* new_node = frozen_graph_def.mutable_node()->Add(); in FreezeRequantizationRanges()
202 return RenameNodeInputs(frozen_graph_def, inputs_to_rename, in FreezeRequantizationRanges()
Dfreeze_requantization_ranges_test.cc122 GraphDef frozen_graph_def; in TestFreezeRequantizationRanges() local
124 FreezeRequantizationRanges(graph_def, context, &frozen_graph_def)); in TestFreezeRequantizationRanges()
127 MapNamesToNodes(frozen_graph_def, &node_map); in TestFreezeRequantizationRanges()
146 TF_ASSERT_OK(frozen_session->Create(frozen_graph_def)); in TestFreezeRequantizationRanges()
/external/tensorflow/tensorflow/examples/speech_commands/
Dfreeze.py161 def save_graph_def(file_name, frozen_graph_def): argument
169 frozen_graph_def,
234 frozen_graph_def = graph_util.convert_variables_to_constants(
238 save_graph_def(FLAGS.output_file, frozen_graph_def)
/external/tensorflow/tensorflow/lite/python/
Dconvert_saved_model.py206 frozen_graph_def = util.freeze_graph(sess, in_tensors, out_tensors)
207 return frozen_graph_def, in_tensors, out_tensors, sess.graph
Dlite_test.py124 frozen_graph_def = (
132 converter = lite.TFLiteConverter(frozen_graph_def, [in_tensor],
154 frozen_graph_def = (
159 converter = lite.TFLiteConverter(frozen_graph_def, None, None,
192 frozen_graph_def = (
197 converter = lite.TFLiteConverter(frozen_graph_def, None, None,
211 frozen_graph_def = (
216 converter = lite.TFLiteConverter(frozen_graph_def, None, None,
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py561 frozen_graph_def = graph_util.convert_variables_to_constants(
565 self._grappler_meta_graph_def.graph_def.CopyFrom(frozen_graph_def)