Home
last modified time | relevance | path

Searched refs:current_graph_def (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dremove_nodes.cc53 GraphDef current_graph_def = input_graph_def; in RemoveNodes() local
69 current_graph_def, pattern, in RemoveNodes()
104 std::unordered_set<string>(), &current_graph_def)); in RemoveNodes()
109 *output_graph_def = current_graph_def; in RemoveNodes()
Dfold_old_batch_norms.cc288 GraphDef current_graph_def = input_graph_def; in FoldOldBatchNorms() local
296 current_graph_def, // clang-format off in FoldOldBatchNorms()
320 current_graph_def = replaced_graph_def; in FoldOldBatchNorms()
327 current_graph_def, // clang-format off in FoldOldBatchNorms()
357 current_graph_def = replaced_graph_def; in FoldOldBatchNorms()
365 current_graph_def, // clang-format off in FoldOldBatchNorms()
400 current_graph_def = replaced_graph_def; in FoldOldBatchNorms()
403 *output_graph_def = current_graph_def; in FoldOldBatchNorms()
Dsparsify_gather.cc234 GraphDef current_graph_def = input_graph_def; in SparsifyGatherInternal() local
239 for (const auto& node : current_graph_def.node()) { in SparsifyGatherInternal()
251 GraphDef replaced_graph_def = current_graph_def; in SparsifyGatherInternal()
256 current_graph_def, pattern, in SparsifyGatherInternal()
555 current_graph_def = replaced_graph_def; in SparsifyGatherInternal()
557 *output_graph_def = current_graph_def; in SparsifyGatherInternal()
Dquantize_nodes.cc200 GraphDef current_graph_def = input_graph_def; in MergeDuplicateNodes() local
207 for (const NodeDef& node : current_graph_def.node()) { in MergeDuplicateNodes()
245 &current_graph_def)); in MergeDuplicateNodes()
248 *output_graph_def = current_graph_def; in MergeDuplicateNodes()
536 GraphDef current_graph_def = input_graph_def; in HoistFakeQuants() local
546 current_graph_def, pattern, in HoistFakeQuants()
579 current_graph_def = hoisted_graph_def; in HoistFakeQuants()
581 *output_graph_def = current_graph_def; in HoistFakeQuants()