Home
last modified time | relevance | path

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

123

/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.py105 output_graph_def = graph_pb2.GraphDef()
107 output_graph_def.ParseFromString(f.read())
108 _ = importer.import_graph_def(output_graph_def, name="")
110 self.assertEqual(4, len(output_graph_def.node))
111 for node in output_graph_def.node:
206 output_graph_def = graph_pb2.GraphDef()
208 output_graph_def.ParseFromString(f.read())
209 _ = importer.import_graph_def(output_graph_def, name="")
211 self.assertEqual(4, len(output_graph_def.node))
212 for node in output_graph_def.node:
[all …]
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(
96 f.write(output_graph_def.SerializeToString())
98 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.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/python/compiler/tensorrt/
Dtrt_convert_test.py196 output_graph_def = conversion_result.graph.as_graph_def()
198 output_graph_def = conversion_result
211 output_graph_def = converter.calibrate(
218 return output_graph_def
226 output_graph_def = self._ConvertGraph(
232 graph_defs_to_verify = [output_graph_def]
359 output_graph_def = self._ConvertGraph(minimum_segment_size=5)
360 node_name_to_op = {node.name: node.op for node in output_graph_def.node}
379 output_graph_def = self._ConvertGraph(
388 importer.import_graph_def(output_graph_def, name="")
[all …]
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfuse_convolutions.cc31 GraphDef* output_graph_def) { in FuseResizePadAndConv() argument
79 *output_graph_def = replaced_graph_def; in FuseResizePadAndConv()
85 GraphDef* output_graph_def) { in FuseResizeAndConv() argument
134 *output_graph_def = replaced_graph_def; in FuseResizeAndConv()
140 GraphDef* output_graph_def) { in FusePadAndConv() argument
188 *output_graph_def = replaced_graph_def; in FusePadAndConv()
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
228 output_graph_def->Clear(); in SortByExecutionOrder()
238 *output_graph_def->mutable_node()->Add() = node_def; in SortByExecutionOrder()
366 const ReplaceMatchingOpTypesOptions& options, GraphDef* output_graph_def) { in ReplaceMatchingOpTypes() argument
[all …]
Dremove_device.cc32 GraphDef* output_graph_def) { in RemoveDevice() argument
33 output_graph_def->Clear(); in RemoveDevice()
35 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()
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.cc32 GraphDef* output_graph_def) { in RenameOp() argument
44 output_graph_def->Clear(); in RenameOp()
46 NodeDef* new_node = output_graph_def->mutable_node()->Add(); in RenameOp()
D__init__.py52 output_graph_def = graph_pb2.GraphDef()
53 output_graph_def.ParseFromString(output_graph_def_string)
54 return output_graph_def
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 …]
Dremove_attribute.cc32 GraphDef* output_graph_def) { in RemoveAttribute() argument
53 output_graph_def->Clear(); in RemoveAttribute()
55 NodeDef* new_node = output_graph_def->mutable_node()->Add(); in RemoveAttribute()
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()
Dbackports.cc32 GraphDef* output_graph_def) { in BackportConcatV2Transform() argument
56 {true}, output_graph_def)); in BackportConcatV2Transform()
66 GraphDef* output_graph_def) { in BackportTensorArrayV3Transform() argument
127 {true}, output_graph_def)); in BackportTensorArrayV3Transform()
Drename_attribute.cc31 GraphDef* output_graph_def) { in RenameAttribute() argument
51 output_graph_def->Clear(); in RenameAttribute()
53 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.cc33 GraphDef* output_graph_def) { in ObfuscateNames() argument
72 output_graph_def->Clear(); in ObfuscateNames()
74 NodeDef* node = output_graph_def->mutable_node()->Add(); in ObfuscateNames()
Dsort_by_execution_order.cc34 GraphDef* output_graph_def) { in SortByExecutionOrderWithUnusedContext() argument
35 return SortByExecutionOrder(input_graph_def, output_graph_def); in SortByExecutionOrderWithUnusedContext()
Dstrip_unused_nodes.cc119 GraphDef* output_graph_def) { in StripUnusedNodes() argument
168 output_graph_def->Clear(); in StripUnusedNodes()
184 *(output_graph_def->mutable_node()->Add()) = placeholder_node; in StripUnusedNodes()
186 *(output_graph_def->mutable_node()->Add()) = node; in StripUnusedNodes()
Dfold_constants_lib.h32 GraphDef* output_graph_def);
38 GraphDef* output_graph_def);

123