Home
last modified time | relevance | path

Searched refs:replaced_graph_def (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dsparsify_gather.cc251 GraphDef replaced_graph_def = current_graph_def; in SparsifyGatherInternal() local
450 {true}, &replaced_graph_def)); in SparsifyGatherInternal()
453 for (int i = 0; i < replaced_graph_def.node_size(); i++) { in SparsifyGatherInternal()
454 if (replaced_graph_def.node(i).name() == group_init_node && in SparsifyGatherInternal()
455 replaced_graph_def.node(i).op() == "NoOp") { in SparsifyGatherInternal()
456 init_op = replaced_graph_def.mutable_node(i); in SparsifyGatherInternal()
462 init_op = replaced_graph_def.mutable_node()->Add(); in SparsifyGatherInternal()
493 while (i < replaced_graph_def.node_size()) { in SparsifyGatherInternal()
495 if ((replaced_graph_def.node(i).name() == name) && in SparsifyGatherInternal()
496 (replaced_graph_def.node(i).op() != "RestoreV2")) { in SparsifyGatherInternal()
[all …]
Dfuse_convolutions.cc32 GraphDef replaced_graph_def; in FuseResizePadAndConv() local
78 {}, &replaced_graph_def)); in FuseResizePadAndConv()
79 *output_graph_def = replaced_graph_def; in FuseResizePadAndConv()
86 GraphDef replaced_graph_def; in FuseResizeAndConv() local
133 {}, &replaced_graph_def)); in FuseResizeAndConv()
134 *output_graph_def = replaced_graph_def; in FuseResizeAndConv()
141 GraphDef replaced_graph_def; in FusePadAndConv() local
187 {}, &replaced_graph_def)); in FusePadAndConv()
188 *output_graph_def = replaced_graph_def; in FusePadAndConv()
Dfold_old_batch_norms.cc294 GraphDef replaced_graph_def; in FoldOldBatchNorms() local
319 {}, &replaced_graph_def)); in FoldOldBatchNorms()
320 current_graph_def = replaced_graph_def; in FoldOldBatchNorms()
325 GraphDef replaced_graph_def; in FoldOldBatchNorms() local
356 {}, &replaced_graph_def)); in FoldOldBatchNorms()
357 current_graph_def = replaced_graph_def; in FoldOldBatchNorms()
362 GraphDef replaced_graph_def; in FoldOldBatchNorms() local
399 {}, &replaced_graph_def)); in FoldOldBatchNorms()
400 current_graph_def = replaced_graph_def; in FoldOldBatchNorms()
Dfold_batch_norms.cc35 GraphDef replaced_graph_def; in FoldBatchNorms() local
118 {}, &replaced_graph_def)); in FoldBatchNorms()
119 *output_graph_def = replaced_graph_def; in FoldBatchNorms()
Dremove_nodes.cc67 GraphDef replaced_graph_def; in RemoveNodes() local
100 {true}, &replaced_graph_def)); in RemoveNodes()
103 RenameNodeInputs(replaced_graph_def, inputs_to_rename, in RemoveNodes()
Dflatten_atrous.cc29 GraphDef replaced_graph_def; in FlattenAtrousConv() local
132 {}, &replaced_graph_def)); in FlattenAtrousConv()
133 *output_graph_def = replaced_graph_def; in FlattenAtrousConv()
Dbackports.cc68 GraphDef replaced_graph_def; in BackportTensorArrayV3Transform() local
106 {true}, &replaced_graph_def)); in BackportTensorArrayV3Transform()
110 TF_RETURN_IF_ERROR(RenameNodeInputs(replaced_graph_def, inputs_to_rename, in BackportTensorArrayV3Transform()
Dfuse_quantized_convolution.cc39 GraphDef replaced_graph_def; in FuseQuantizedConvolutionAndRequantize() local
137 {}, &replaced_graph_def)); in FuseQuantizedConvolutionAndRequantize()
147 MapNamesToNodes(replaced_graph_def, &node_map); in FuseQuantizedConvolutionAndRequantize()
216 *output_graph_def = replaced_graph_def; in FuseQuantizedConvolutionAndRequantize()
Dquantize_nodes.cc269 GraphDef replaced_graph_def; in RemoveRedundantQuantizations() local
302 {true}, &replaced_graph_def)); in RemoveRedundantQuantizations()
304 return RenameNodeInputs(replaced_graph_def, inputs_to_rename, in RemoveRedundantQuantizations()
Dtransform_utils_test.cc432 GraphDef replaced_graph_def; in TestReplaceMatchingOpTypes() local
452 {}, &replaced_graph_def)); in TestReplaceMatchingOpTypes()
454 EXPECT_EQ(10, replaced_graph_def.node_size()); in TestReplaceMatchingOpTypes()
455 for (const NodeDef& node : replaced_graph_def.node()) { in TestReplaceMatchingOpTypes()
DREADME.md1007 {}, &replaced_graph_def));