Home
last modified time | relevance | path

Searched refs:input_graph_def (Results 1 – 25 of 76) sorted by relevance

1234

/external/tensorflow/tensorflow/python/tools/
Dfreeze_graph.py77 def freeze_graph_with_def_protos(input_graph_def, argument
141 elif input_graph_def:
142 for node in input_graph_def.node:
145 if input_graph_def:
146 _ = importer.import_graph_def(input_graph_def, name="")
229 input_graph_def,
247 input_graph_def = graph_pb2.GraphDef()
251 input_graph_def.ParseFromString(f.read())
253 text_format.Merge(f.read(), input_graph_def)
254 return input_graph_def
[all …]
Dstrip_unused_lib.py32 def strip_unused(input_graph_def, input_node_names, output_node_names, argument
61 for node in input_graph_def.node:
103 input_graph_def = graph_pb2.GraphDef()
107 input_graph_def.ParseFromString(f.read())
109 text_format.Merge(f.read(), input_graph_def)
111 output_graph_def = strip_unused(input_graph_def,
Doptimize_for_inference_lib.py89 def optimize_for_inference(input_graph_def, input_node_names, output_node_names, argument
107 ensure_graph_is_valid(input_graph_def)
108 optimized_graph_def = input_graph_def
204 def fold_batch_norms(input_graph_def): argument
228 for node in input_graph_def.node:
236 for node in input_graph_def.node:
361 for node in input_graph_def.node:
372 def fuse_resize_and_conv(input_graph_def, output_node_names): argument
393 for node in input_graph_def.node:
400 for node in input_graph_def.node:
[all …]
Doptimize_for_inference.py79 input_graph_def = graph_pb2.GraphDef()
83 input_graph_def.ParseFromString(data)
85 text_format.Merge(data.decode("utf-8"), input_graph_def)
88 input_graph_def,
/external/tensorflow/tensorflow/tools/graph_transforms/python/
Dtransform_graph_test.py35 input_graph_def = graph_pb2.GraphDef()
37 const_op1 = input_graph_def.node.add()
46 const_op2 = input_graph_def.node.add()
56 add_op = input_graph_def.node.add()
64 relu_op = input_graph_def.node.add()
76 transformed_graph_def = TransformGraph(input_graph_def, input_names,
/external/tensorflow/tensorflow/contrib/model_pruning/python/
Dstrip_pruning_vars_lib.py53 def _get_masked_weights(input_graph_def): argument
57 importer.import_graph_def(input_graph_def, name='')
61 for node in input_graph_def.node:
74 def strip_pruning_vars_fn(input_graph_def, output_node_names): argument
89 masked_weights_dict = _get_masked_weights(input_graph_def)
94 for node in input_graph_def.node:
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfuse_convolutions.cc29 Status FuseResizePadAndConv(const GraphDef& input_graph_def, in FuseResizePadAndConv() argument
34 input_graph_def, // clang-format off in FuseResizePadAndConv()
83 Status FuseResizeAndConv(const GraphDef& input_graph_def, in FuseResizeAndConv() argument
88 input_graph_def, // clang-format off in FuseResizeAndConv()
138 Status FusePadAndConv(const GraphDef& input_graph_def, in FusePadAndConv() argument
143 input_graph_def, // clang-format off in FusePadAndConv()
Dtransform_utils.cc158 void FilterGraphDef(const GraphDef& input_graph_def, in FilterGraphDef() argument
162 for (const NodeDef& node : input_graph_def.node()) { in FilterGraphDef()
169 void RemoveAttributes(const GraphDef& input_graph_def, in RemoveAttributes() argument
173 for (const NodeDef& node : input_graph_def.node()) { in RemoveAttributes()
182 Status SortByExecutionOrder(const GraphDef& input_graph_def, in SortByExecutionOrder() argument
184 const int num_nodes = input_graph_def.node_size(); in SortByExecutionOrder()
191 for (int i = 0; i < input_graph_def.node_size(); ++i) { in SortByExecutionOrder()
192 const NodeDef& node(input_graph_def.node(i)); in SortByExecutionOrder()
198 const NodeDef& node_def(input_graph_def.node(n)); in SortByExecutionOrder()
237 const NodeDef& node_def(input_graph_def.node(o)); in SortByExecutionOrder()
[all …]
Dadd_default_attributes.cc24 Status AddDefaultAttributes(const GraphDef& input_graph_def, in AddDefaultAttributes() argument
30 input_graph_def.library())); in AddDefaultAttributes()
32 *output_graph_def = input_graph_def; in AddDefaultAttributes()
Dsparsify_gather.cc98 Status ObtainTensorSlice(const GraphDef& input_graph_def, in ObtainTensorSlice() argument
102 for (const auto& node : input_graph_def.node()) { in ObtainTensorSlice()
117 for (const auto& node : input_graph_def.node()) { in ObtainTensorSlice()
126 for (const auto& node : input_graph_def.node()) { in ObtainTensorSlice()
143 for (const auto& node : input_graph_def.node()) { in ObtainTensorSlice()
194 const GraphDef& input_graph_def, in ObtainVariableInfo() argument
197 for (const auto& node : input_graph_def.node()) { in ObtainVariableInfo()
200 TF_RETURN_IF_ERROR(ObtainTensorSlice(input_graph_def, node.name(), &s)); in ObtainVariableInfo()
224 const GraphDef& input_graph_def, in SparsifyGatherInternal() argument
234 GraphDef current_graph_def = input_graph_def; in SparsifyGatherInternal()
[all …]
Dfold_constants_lib.cc144 Status RemoveUnusedNodes(const GraphDef& input_graph_def, in RemoveUnusedNodes() argument
148 for (const NodeDef& node : input_graph_def.node()) { in RemoveUnusedNodes()
189 input_graph_def, in RemoveUnusedNodes()
213 Status FoldConstants(const GraphDef& input_graph_def, in FoldConstants() argument
217 TF_RETURN_IF_ERROR(input_graph.AddFunctionLibrary(input_graph_def.library())); in FoldConstants()
232 RemoveAttributes(input_graph_def, {"_output_shapes"}, &cleaned_graph_def); in FoldConstants()
238 ImportGraphDef({}, input_graph_def, &input_graph, &shape_refiner)); in FoldConstants()
328 TF_RETURN_IF_ERROR(ReplaceSendRecvs(input_graph_def, folded_graph_def, in FoldConstants()
Dbackports.cc30 Status BackportConcatV2Transform(const GraphDef& input_graph_def, in BackportConcatV2Transform() argument
34 input_graph_def, {"ConcatV2"}, in BackportConcatV2Transform()
64 Status BackportTensorArrayV3Transform(const GraphDef& input_graph_def, in BackportTensorArrayV3Transform() argument
70 input_graph_def, {"TensorArrayV3|TensorArrayGradV3"}, in BackportTensorArrayV3Transform()
Dquantize_nodes.cc192 Status MergeDuplicateNodes(const GraphDef& input_graph_def, in MergeDuplicateNodes() argument
200 GraphDef current_graph_def = input_graph_def; in MergeDuplicateNodes()
261 Status RemoveRedundantQuantizations(const GraphDef& input_graph_def, in RemoveRedundantQuantizations() argument
271 input_graph_def, // clang-format off in RemoveRedundantQuantizations()
311 Status QuantizePlaceholders(const GraphDef& input_graph_def, in QuantizePlaceholders() argument
321 *output_graph_def = input_graph_def; in QuantizePlaceholders()
328 for (const NodeDef& node : input_graph_def.node()) { in QuantizePlaceholders()
391 Status ConvertFakeQuantsToRequantize(const GraphDef& input_graph_def, in ConvertFakeQuantsToRequantize() argument
395 input_graph_def, // clang-format off in ConvertFakeQuantsToRequantize()
463 Status MergeAdjacentRequantizes(const GraphDef& input_graph_def, in MergeAdjacentRequantizes() argument
[all …]
Dtransform_utils.h108 void FilterGraphDef(const GraphDef& input_graph_def,
114 void RemoveAttributes(const GraphDef& input_graph_def,
121 Status SortByExecutionOrder(const GraphDef& input_graph_def,
199 const GraphDef& input_graph_def, const OpTypePattern& pattern,
210 Status RenameNodeInputs(const GraphDef& input_graph_def,
Dobfuscate_names.cc31 Status ObfuscateNames(const GraphDef& input_graph_def, in ObfuscateNames() argument
48 for (const NodeDef& input_node : input_graph_def.node()) { in ObfuscateNames()
73 for (const NodeDef& input_node : input_graph_def.node()) { in ObfuscateNames()
Dsort_by_execution_order.cc33 const GraphDef& input_graph_def, const TransformFuncContext& unused_context, in SortByExecutionOrderWithUnusedContext() argument
35 return SortByExecutionOrder(input_graph_def, output_graph_def); in SortByExecutionOrderWithUnusedContext()
Dfold_constants_lib.h30 Status FoldConstants(const GraphDef& input_graph_def,
36 Status RemoveUnusedNodes(const GraphDef& input_graph_def,
Dremove_device.cc30 Status RemoveDevice(const GraphDef& input_graph_def, in RemoveDevice() argument
34 for (const NodeDef& node : input_graph_def.node()) { in RemoveDevice()
Dremove_control_dependencies.cc26 Status RemoveControlDependencies(const GraphDef& input_graph_def, in RemoveControlDependencies() argument
30 for (const NodeDef& node : input_graph_def.node()) { in RemoveControlDependencies()
Dset_device.cc23 Status SetDevice(const GraphDef& input_graph_def, in SetDevice() argument
33 for (const NodeDef& node : input_graph_def.node()) { in SetDevice()
Drename_op.cc30 Status RenameOp(const GraphDef& input_graph_def, in RenameOp() argument
45 for (const NodeDef& node : input_graph_def.node()) { in RenameOp()
/external/tensorflow/tensorflow/contrib/tensorrt/python/
Dtrt_convert.py25 input_graph_def, argument
39 input_graph_def=input_graph_def,
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.h219 static Status FuseCluster(const GraphDef& input_graph_def,
230 const GraphDef& input_graph_def, const std::vector<string>& inputs,
239 const GraphDef& input_graph_def, const std::vector<string>& inputs,
249 const GraphDef& input_graph_def, const std::vector<string>& inputs,
268 const GraphDef& input_graph_def,
272 static Status FuseRemoteGraphByExecutor(const GraphDef& input_graph_def,
279 const GraphDef& input_graph_def,
Dremote_fused_graph_execute_utils.cc943 const GraphDef& input_graph_def, const std::vector<string>& inputs, in FuseCluster() argument
958 BuildClusterSubgraphDef(cluster, input_graph_def, &subgraph_def)); in FuseCluster()
963 ImportGraphDef({}, input_graph_def, &graph, &shape_refiner)); in FuseCluster()
1035 const GraphDef& input_graph_def, const std::vector<string>& inputs, in FuseRemoteGraphByNodeNames() argument
1043 subgraph_nodes, input_graph_def, &ci_vec)); in FuseRemoteGraphByNodeNames()
1048 TF_RETURN_IF_ERROR(FuseCluster(input_graph_def, inputs, outputs, in FuseRemoteGraphByNodeNames()
1057 const GraphDef& input_graph_def, const std::vector<string>& inputs, in FuseRemoteGraphByBorder() argument
1066 border_inputs, border_outputs, input_graph_def, &cluster)); in FuseRemoteGraphByBorder()
1069 input_graph_def, inputs, outputs, remote_fused_graph_node_name, cluster, in FuseRemoteGraphByBorder()
1074 const GraphDef& input_graph_def, const std::vector<string>& inputs, in FuseRemoteGraphByOpTypes() argument
[all …]
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py107 input_graph_def=None, argument
131 if input_graph_def or not input_saved_model_dir:
136 if input_graph_def and input_saved_model_dir:
139 if not input_graph_def and not input_saved_model_dir:
143 self._input_graph_def = input_graph_def
561 input_graph_def=None, argument
629 input_graph_def=input_graph_def,
782 input_graph_def, argument
862 input_graph_def=input_graph_def,

1234