Home
last modified time | relevance | path

Searched refs:output_graph_def (Results 1 – 25 of 79) sorted by relevance

1234

/external/tensorflow/tensorflow/python/tools/
Dstrip_unused_test.py84 output_graph_def = graph_pb2.GraphDef()
86 output_graph_def.ParseFromString(f.read())
87 _ = importer.import_graph_def(output_graph_def, name="")
89 self.assertEqual(3, len(output_graph_def.node))
90 for node in output_graph_def.node:
141 output_graph_def = graph_pb2.GraphDef()
143 output_graph_def.ParseFromString(f.read())
144 _ = importer.import_graph_def(output_graph_def, name="")
146 self.assertEqual(3, len(output_graph_def.node))
147 for node in output_graph_def.node:
Dfreeze_graph_test.py107 output_graph_def = graph_pb2.GraphDef()
109 output_graph_def.ParseFromString(f.read())
110 _ = importer.import_graph_def(output_graph_def, name="")
112 self.assertEqual(4, len(output_graph_def.node))
113 for node in output_graph_def.node:
209 output_graph_def = graph_pb2.GraphDef()
211 output_graph_def.ParseFromString(f.read())
212 _ = importer.import_graph_def(output_graph_def, name="")
214 self.assertEqual(4, len(output_graph_def.node))
215 for node in output_graph_def.node:
[all …]
Dstrip_unused_lib.py87 output_graph_def = graph_util.extract_sub_graph(inputs_replaced_graph_def,
89 return output_graph_def
113 output_graph_def = strip_unused(input_graph_def,
120 f.write(output_graph_def.SerializeToString())
123 f.write(text_format.MessageToString(output_graph_def))
124 print("%d ops in the final graph." % len(output_graph_def.node))
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_rewriter_transform.cc140 GraphDef* output_graph_def) { in FuseRemoteGraph() argument
176 output_graph_def)); in FuseRemoteGraph()
191 require_shape_type, output_graph_def)); in FuseRemoteGraph()
200 output_graph_def)); in FuseRemoteGraph()
204 output_graph_def)); in FuseRemoteGraph()
214 GraphDef* output_graph_def) { in PlaceRemoteGraphArguments() argument
215 *output_graph_def = input_graph_def; in PlaceRemoteGraphArguments()
237 input_shapes_str, output_graph_def)); in PlaceRemoteGraphArguments()
256 output_graph_def)); in PlaceRemoteGraphArguments()
Dremote_fused_graph_execute_utils.h226 GraphDef* output_graph_def);
235 const bool require_shape_type, GraphDef* output_graph_def);
245 GraphDef* output_graph_def);
254 const bool require_shape_type, GraphDef* output_graph_def);
270 GraphDef* output_graph_def);
276 GraphDef* output_graph_def);
/external/tensorflow/tensorflow/tools/graph_transforms/
Dremove_device.cc31 GraphDef* output_graph_def) { in RemoveDevice() argument
32 output_graph_def->Clear(); in RemoveDevice()
34 NodeDef* new_node = output_graph_def->mutable_node()->Add(); in RemoveDevice()
Dadd_default_attributes.cc26 GraphDef* output_graph_def) { in AddDefaultAttributes() argument
32 *output_graph_def = input_graph_def; in AddDefaultAttributes()
33 TF_RETURN_IF_ERROR(AddDefaultAttrsToGraphDef(output_graph_def, *flib_def, 0)); in AddDefaultAttributes()
Dremove_control_dependencies.cc28 GraphDef* output_graph_def) { in RemoveControlDependencies() argument
29 output_graph_def->Clear(); in RemoveControlDependencies()
31 NodeDef* new_node = output_graph_def->mutable_node()->Add(); in RemoveControlDependencies()
Dfuse_convolutions.cc30 GraphDef* output_graph_def) { in FuseResizePadAndConv() argument
78 *output_graph_def = replaced_graph_def; in FuseResizePadAndConv()
84 GraphDef* output_graph_def) { in FuseResizeAndConv() argument
133 *output_graph_def = replaced_graph_def; in FuseResizeAndConv()
139 GraphDef* output_graph_def) { in FusePadAndConv() argument
187 *output_graph_def = replaced_graph_def; in FusePadAndConv()
Dset_device.cc25 GraphDef* output_graph_def) { in SetDevice() argument
32 output_graph_def->Clear(); in SetDevice()
34 NodeDef* new_node = output_graph_def->mutable_node()->Add(); in SetDevice()
D__init__.py49 output_graph_def = graph_pb2.GraphDef()
50 output_graph_def.ParseFromString(output_graph_def_string)
51 return output_graph_def
Drename_op.cc31 GraphDef* output_graph_def) { in RenameOp() argument
43 output_graph_def->Clear(); in RenameOp()
45 NodeDef* new_node = output_graph_def->mutable_node()->Add(); in RenameOp()
Dtransform_utils.cc162 GraphDef* output_graph_def) { in FilterGraphDef() argument
163 output_graph_def->mutable_node()->Clear(); in FilterGraphDef()
166 *output_graph_def->mutable_node()->Add() = node; in FilterGraphDef()
173 GraphDef* output_graph_def) { in RemoveAttributes() argument
174 output_graph_def->mutable_node()->Clear(); in RemoveAttributes()
176 NodeDef* new_node = output_graph_def->mutable_node()->Add(); in RemoveAttributes()
185 GraphDef* output_graph_def) { in SortByExecutionOrder() argument
230 output_graph_def->Clear(); in SortByExecutionOrder()
240 *output_graph_def->mutable_node()->Add() = node_def; in SortByExecutionOrder()
368 const ReplaceMatchingOpTypesOptions& options, GraphDef* output_graph_def) { in ReplaceMatchingOpTypes() argument
[all …]
Dremove_attribute.cc31 GraphDef* output_graph_def) { in RemoveAttribute() argument
52 output_graph_def->Clear(); in RemoveAttribute()
54 NodeDef* new_node = output_graph_def->mutable_node()->Add(); in RemoveAttribute()
Dinline_partitionedcall.cc59 GraphDef* output_graph_def) { in InlinePartitionedCall() argument
60 output_graph_def->Clear(); in InlinePartitionedCall()
111 NodeDef* new_node = output_graph_def->mutable_node()->Add(); in InlinePartitionedCall()
128 NodeDef* new_node = output_graph_def->mutable_node()->Add(); in InlinePartitionedCall()
134 for (NodeDef& node : *output_graph_def->mutable_node()) { in InlinePartitionedCall()
Drename_attribute.cc30 GraphDef* output_graph_def) { in RenameAttribute() argument
50 output_graph_def->Clear(); in RenameAttribute()
52 NodeDef* new_node = output_graph_def->mutable_node()->Add(); in RenameAttribute()
Dbackports.cc31 GraphDef* output_graph_def) { in BackportConcatV2Transform() argument
55 {true}, output_graph_def)); in BackportConcatV2Transform()
65 GraphDef* output_graph_def) { in BackportTensorArrayV3Transform() argument
126 {true}, output_graph_def)); in BackportTensorArrayV3Transform()
Drename_node.cc27 GraphDef* output_graph_def) { in RenameNode() argument
42 output_graph_def->Clear(); in RenameNode()
44 NodeDef* node = output_graph_def->mutable_node()->Add(); in RenameNode()
Dfold_constants_lib.cc50 GraphDef* output_graph_def) { in ReplaceSendRecvs() argument
80 NodeDef* new_node = output_graph_def->add_node(); in ReplaceSendRecvs()
113 output_graph_def->add_node()->MergeFrom(removed_node); in ReplaceSendRecvs()
146 GraphDef* output_graph_def) { in RemoveUnusedNodes() argument
191 output_graph_def); in RemoveUnusedNodes()
192 TF_RETURN_IF_ERROR(RewriteInputsAsPlaceholders(context, output_graph_def)); in RemoveUnusedNodes()
215 GraphDef* output_graph_def) { in FoldConstants() argument
332 RemoveUnusedNodes(send_recvs_replaced, context, output_graph_def)); in FoldConstants()
Dquantize_nodes.cc194 GraphDef* output_graph_def) { in MergeDuplicateNodes() argument
248 *output_graph_def = current_graph_def; in MergeDuplicateNodes()
263 GraphDef* output_graph_def) { in RemoveRedundantQuantizations() argument
305 std::unordered_set<string>(), output_graph_def); in RemoveRedundantQuantizations()
313 GraphDef* output_graph_def) { in QuantizePlaceholders() argument
321 *output_graph_def = input_graph_def; in QuantizePlaceholders()
383 std::unordered_set<string>(), output_graph_def)); in QuantizePlaceholders()
393 GraphDef* output_graph_def) { in ConvertFakeQuantsToRequantize() argument
452 {}, output_graph_def)); in ConvertFakeQuantsToRequantize()
465 GraphDef* output_graph_def) { in MergeAdjacentRequantizes() argument
[all …]
Dobfuscate_names.cc32 GraphDef* output_graph_def) { in ObfuscateNames() argument
71 output_graph_def->Clear(); in ObfuscateNames()
73 NodeDef* node = output_graph_def->mutable_node()->Add(); in ObfuscateNames()
Dsort_by_execution_order.cc33 GraphDef* output_graph_def) { in SortByExecutionOrderWithUnusedContext() argument
34 return SortByExecutionOrder(input_graph_def, output_graph_def); in SortByExecutionOrderWithUnusedContext()
Dfold_constants_lib.h32 GraphDef* output_graph_def);
38 GraphDef* output_graph_def);
Dtransform_utils.h110 GraphDef* output_graph_def);
116 GraphDef* output_graph_def);
122 GraphDef* output_graph_def);
203 const ReplaceMatchingOpTypesOptions& options, GraphDef* output_graph_def);
213 GraphDef* output_graph_def);
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph_test.cc154 Status RunConvertAfterShape(Scope s, GraphDef* output_graph_def, in RunConvertAfterShape() argument
168 params.output_graph_def = output_graph_def; in RunConvertAfterShape()
205 GraphDef output_graph_def; in TEST_F() local
206 TF_EXPECT_OK(RunConvertAfterShape(s, &output_graph_def)); in TEST_F()
213 for (const NodeDef& node : output_graph_def.node()) { in TEST_F()

1234