Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/python/tools/
Dfreeze_graph_test.py104 output_graph_def = graph_pb2.GraphDef()
106 output_graph_def.ParseFromString(f.read())
107 _ = importer.import_graph_def(output_graph_def, name="")
109 self.assertEqual(4, len(output_graph_def.node))
110 for node in output_graph_def.node:
203 output_graph_def = graph_pb2.GraphDef()
205 output_graph_def.ParseFromString(f.read())
206 _ = importer.import_graph_def(output_graph_def, name="")
208 self.assertEqual(4, len(output_graph_def.node))
209 for node in output_graph_def.node:
[all …]
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:
Dstrip_unused_lib.py85 output_graph_def = graph_util.extract_sub_graph(inputs_replaced_graph_def,
87 return output_graph_def
111 output_graph_def = strip_unused(input_graph_def,
118 f.write(output_graph_def.SerializeToString())
121 f.write(text_format.MessageToString(output_graph_def))
122 print("%d ops in the final graph." % len(output_graph_def.node))
Doptimize_for_inference.py87 output_graph_def = optimize_for_inference_lib.optimize_for_inference(
94 f.write(output_graph_def.SerializeToString())
96 graph_io.write_graph(output_graph_def,
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_rewriter_transform.cc134 GraphDef* output_graph_def) { in FuseRemoteGraph() argument
170 output_graph_def)); in FuseRemoteGraph()
185 require_shape_type, output_graph_def)); in FuseRemoteGraph()
194 output_graph_def)); in FuseRemoteGraph()
198 output_graph_def)); in FuseRemoteGraph()
208 GraphDef* output_graph_def) { in PlaceRemoteGraphArguments() argument
209 *output_graph_def = input_graph_def; in PlaceRemoteGraphArguments()
231 input_shapes_str, output_graph_def)); in PlaceRemoteGraphArguments()
250 output_graph_def)); in PlaceRemoteGraphArguments()
Dremote_fused_graph_execute_utils.h217 GraphDef* output_graph_def);
226 const bool require_shape_type, GraphDef* output_graph_def);
236 GraphDef* output_graph_def);
245 const bool require_shape_type, GraphDef* output_graph_def);
261 GraphDef* output_graph_def);
267 GraphDef* output_graph_def);
/external/tensorflow/tensorflow/tools/graph_transforms/
Dtransform_utils.cc160 GraphDef* output_graph_def) { in FilterGraphDef() argument
161 output_graph_def->mutable_node()->Clear(); in FilterGraphDef()
164 *output_graph_def->mutable_node()->Add() = node; in FilterGraphDef()
171 GraphDef* output_graph_def) { in RemoveAttributes() argument
172 output_graph_def->mutable_node()->Clear(); in RemoveAttributes()
174 NodeDef* new_node = output_graph_def->mutable_node()->Add(); in RemoveAttributes()
183 GraphDef* output_graph_def) { in SortByExecutionOrder() argument
229 output_graph_def->Clear(); in SortByExecutionOrder()
239 *output_graph_def->mutable_node()->Add() = node_def; in SortByExecutionOrder()
360 const ReplaceMatchingOpTypesOptions& options, GraphDef* output_graph_def) { in ReplaceMatchingOpTypes() argument
[all …]
Dfuse_convolutions.cc32 GraphDef* output_graph_def) { in FuseResizePadAndConv() argument
80 *output_graph_def = replaced_graph_def; in FuseResizePadAndConv()
86 GraphDef* output_graph_def) { in FuseResizeAndConv() argument
135 *output_graph_def = replaced_graph_def; in FuseResizeAndConv()
141 GraphDef* output_graph_def) { in FusePadAndConv() argument
189 *output_graph_def = replaced_graph_def; in FusePadAndConv()
Dquantize_nodes.cc195 GraphDef* output_graph_def) { in MergeDuplicateNodes() argument
249 *output_graph_def = current_graph_def; in MergeDuplicateNodes()
264 GraphDef* output_graph_def) { in RemoveRedundantQuantizations() argument
306 std::unordered_set<string>(), output_graph_def); in RemoveRedundantQuantizations()
314 GraphDef* output_graph_def) { in QuantizePlaceholders() argument
322 *output_graph_def = input_graph_def; in QuantizePlaceholders()
384 std::unordered_set<string>(), output_graph_def)); in QuantizePlaceholders()
394 GraphDef* output_graph_def) { in ConvertFakeQuantsToRequantize() argument
453 {}, output_graph_def)); in ConvertFakeQuantsToRequantize()
466 GraphDef* output_graph_def) { in MergeAdjacentRequantizes() argument
[all …]
Dfold_constants_lib.cc51 GraphDef* output_graph_def) { in ReplaceSendRecvs() argument
81 NodeDef* new_node = output_graph_def->add_node(); in ReplaceSendRecvs()
114 output_graph_def->add_node()->MergeFrom(removed_node); in ReplaceSendRecvs()
122 GraphDef* output_graph_def) { in RemoveUnusedNodes() argument
167 output_graph_def); in RemoveUnusedNodes()
190 GraphDef* output_graph_def) { in FoldConstants() argument
297 RemoveUnusedNodes(send_recvs_replaced, context, output_graph_def)); in FoldConstants()
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_device.cc33 GraphDef* output_graph_def) { in RemoveDevice() argument
34 output_graph_def->Clear(); in RemoveDevice()
36 NodeDef* new_node = output_graph_def->mutable_node()->Add(); in RemoveDevice()
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()
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()
Drename_op.cc33 GraphDef* output_graph_def) { in RenameOp() argument
45 output_graph_def->Clear(); in RenameOp()
47 NodeDef* new_node = output_graph_def->mutable_node()->Add(); in RenameOp()
Dbackports.cc33 GraphDef* output_graph_def) { in BackportConcatV2Transform() argument
57 {true}, output_graph_def)); in BackportConcatV2Transform()
67 GraphDef* output_graph_def) { in BackportTensorArrayV3Transform() argument
128 {true}, output_graph_def)); in BackportTensorArrayV3Transform()
D__init__.py52 output_graph_def = graph_pb2.GraphDef()
53 output_graph_def.ParseFromString(output_graph_def_string)
54 return output_graph_def
Dremove_attribute.cc33 GraphDef* output_graph_def) { in RemoveAttribute() argument
54 output_graph_def->Clear(); in RemoveAttribute()
56 NodeDef* new_node = output_graph_def->mutable_node()->Add(); in RemoveAttribute()
Drename_attribute.cc32 GraphDef* output_graph_def) { in RenameAttribute() argument
52 output_graph_def->Clear(); in RenameAttribute()
54 NodeDef* new_node = output_graph_def->mutable_node()->Add(); in RenameAttribute()
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);
Dobfuscate_names.cc34 GraphDef* output_graph_def) { in ObfuscateNames() argument
73 output_graph_def->Clear(); in ObfuscateNames()
75 NodeDef* node = output_graph_def->mutable_node()->Add(); in ObfuscateNames()
Dstrip_unused_nodes.cc120 GraphDef* output_graph_def) { in StripUnusedNodes() argument
169 output_graph_def->Clear(); in StripUnusedNodes()
185 *(output_graph_def->mutable_node()->Add()) = placeholder_node; in StripUnusedNodes()
187 *(output_graph_def->mutable_node()->Add()) = node; in StripUnusedNodes()
Dremove_ema_test.cc30 GraphDef* output_graph_def);
34 GraphDef* output_graph_def);
38 GraphDef* output_graph_def);
Dsort_by_execution_order.cc35 GraphDef* output_graph_def) { in SortByExecutionOrderWithUnusedContext() argument
36 return SortByExecutionOrder(input_graph_def, output_graph_def); in SortByExecutionOrderWithUnusedContext()
/external/tensorflow/tensorflow/contrib/tensorrt/python/
Dtrt_convert.py100 output_graph_def = graph_pb2.GraphDef()
101 output_graph_def.ParseFromString(output_graph_def_string)
103 return output_graph_def

123