Home
last modified time | relevance | path

Searched refs:output_node_names (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/contrib/model_pruning/python/
Dstrip_pruning_vars.py43 def strip_pruning_vars(checkpoint_dir, output_node_names, output_dir, filename): argument
61 if not output_node_names:
64 output_node_names = output_node_names.replace(' ', '').split(',')
67 checkpoint_dir, output_node_names)
70 initial_graph_def, output_node_names)
77 return strip_pruning_vars(FLAGS.checkpoint_dir, FLAGS.output_node_names,
Dstrip_pruning_vars_lib.py74 def strip_pruning_vars_fn(input_graph_def, output_node_names): argument
110 return graph_util.extract_sub_graph(pruned_graph_def, output_node_names)
113 def graph_def_from_checkpoint(checkpoint_dir, output_node_names): argument
140 sess, graph_def, output_node_names)
/external/tensorflow/tensorflow/python/tools/
Dstrip_unused_lib.py32 def strip_unused(input_graph_def, input_node_names, output_node_names, argument
86 output_node_names)
91 output_binary, input_node_names, output_node_names, argument
99 if not output_node_names:
113 output_node_names.split(","),
Dfreeze_graph_test.py81 output_node_names = "output_node"
92 output_node_names,
192 output_node_names = "output_node"
200 output_node_names, restore_op_name, filename_tensor_name,
229 output_node_names = "output_node"
241 input_binary, checkpoint_path, output_node_names,
316 output_node_names = "save/restore_all"
323 output_node_names=output_node_names,
Dfreeze_graph.py80 output_node_names, argument
131 if not output_node_names:
223 output_node_names.replace(" ", "").split(","),
230 output_node_names.replace(" ", "").split(","),
292 output_node_names, argument
352 output_node_names,
376 flags.input_checkpoint, flags.output_node_names,
Dstrip_unused_test.py62 output_node_names = "output_node"
69 output_node_names,
129 output_node_names = "output_node"
135 output_node_names,
Doptimize_for_inference_lib.py89 def optimize_for_inference(input_graph_def, input_node_names, output_node_names, argument
110 optimized_graph_def, input_node_names, output_node_names,
113 optimized_graph_def, output_node_names)
117 output_node_names)
372 def fuse_resize_and_conv(input_graph_def, output_node_names): argument
404 for output_name in output_node_names:
Dstrip_unused.py60 FLAGS.output_node_names,
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transferer.h63 const std::vector<string>& output_node_names,
73 const std::vector<string>& output_node_names, const bool is_text_proto,
79 void SortParams(const std::vector<string>& output_node_names);
122 const std::vector<string>& output_node_names);
173 const std::vector<string>& output_node_names);
Dgraph_transferer_test.cc307 const std::vector<string> output_node_names = {NAME_A_PLUS_B}; in TEST_F() local
309 inputs, output_node_names, false); in TEST_F()
318 const std::vector<string> output_node_names = {"softmax"}; in TEST_F() local
320 input_node_info_list, output_node_names, in TEST_F()
344 const std::vector<string> output_node_names = {"softmax"}; in TEST_F() local
346 input_node_info_list, output_node_names, in TEST_F()
379 std::vector<string> output_node_names = {}; in TEST() local
393 *ops_definitions, filename, input_node_info_list, output_node_names, in TEST()
442 std::vector<string> output_node_names = {}; in TEST() local
459 *ops_definitions, filename, input_node_info_list, output_node_names, in TEST()
[all …]
Dhexagon_graph_execution_test.cc306 std::vector<string> output_node_names; in RunFusedGraph() local
307 output_node_names.emplace_back(REMOTE_FUSED_GRAPH_EXECUTE_NODE_NAME); in RunFusedGraph()
311 TF_ASSERT_OK(session->Run(run_options, input_tensors, output_node_names, {}, in RunFusedGraph()
413 std::vector<string> output_node_names = {"softmax"}; in TEST() local
421 *ops_definitions, MODEL_FILENAME, inputs, output_node_names, in TEST()
446 std::vector<string> output_node_names = {"softmax"}; in TEST() local
455 output_node_names, in TEST()
480 std::vector<string> output_node_names = {"softmax"}; in TEST() local
488 *ops_definitions, MODEL_FILENAME, inputs, output_node_names, in TEST()
549 std::vector<string> output_node_names = {"softmax"}; in TEST() local
[all …]
Dgraph_transferer.cc95 const std::vector<string>& output_node_names, in LoadGraphFromProto() argument
142 output_node_names); in LoadGraphFromProto()
149 SortParams(output_node_names); in LoadGraphFromProto()
162 for (const string& output_node_name : output_node_names) { in LoadGraphFromProto()
202 const std::vector<string>& output_node_names, const bool is_text_proto, in LoadGraphFromProtoFile() argument
236 output_node_names, in LoadGraphFromProtoFile()
240 void GraphTransferer::SortParams(const std::vector<string>& output_node_names) { in SortParams() argument
254 for (const string& output_node_name : output_node_names) { in SortParams()
425 const std::vector<string>& output_node_names) { in RegisterNode() argument
843 const std::vector<string>& output_node_names) { in RegisterNodeIfAllInputsAreCached() argument
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.cc204 const std::vector<string>& output_node_names, const bool initialize_by_zero, in DryRunInference() argument
258 status = session->Run(run_options, input_tensors, output_node_names, {}, in DryRunInference()
275 std::vector<string> output_node_names; in DryRunInferenceForAllNode() local
288 output_node_names.emplace_back(strings::StrCat(node->name(), ":", i)); in DryRunInferenceForAllNode()
292 status = DryRunInference(graph_def, input_node_info_list, output_node_names, in DryRunInferenceForAllNode()
299 CHECK_EQ(output_node_names.size(), output_tensors.size()) in DryRunInferenceForAllNode()
300 << output_node_names.size() << ", " << output_tensors.size(); in DryRunInferenceForAllNode()
309 for (int i = 0; static_cast<size_t>(i) < output_node_names.size(); ++i) { in DryRunInferenceForAllNode()
310 const string& name = output_node_names.at(i); in DryRunInferenceForAllNode()
316 const Tensor& tensor = output_tensors.at(output_node_names.size() + i); in DryRunInferenceForAllNode()
[all …]
Dremote_fused_graph_execute_utils.h108 const std::vector<string>& output_node_names,
127 const std::vector<string>& output_node_names,
/external/tensorflow/tensorflow/contrib/meta_graph_transform/
Dmeta_graph_transform.py308 output_node_names = [_op_name(x) for x in output_names]
310 transformed_graph_def, output_node_names,
350 def _freeze_graph_with_def_protos(input_graph_def, output_node_names, argument
381 sess, input_graph_def, output_node_names + initializer_names)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.graph_util.pbtxt5 …argspec: "args=[\'sess\', \'input_graph_def\', \'output_node_names\', \'variable_names_whitelist\'…
/external/tensorflow/tensorflow/python/framework/
Dgraph_util_impl.py223 output_node_names, argument
247 inference_graph = extract_sub_graph(input_graph_def, output_node_names)
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py231 output_node_names = _gather_names(input_signature_def.inputs).union(
238 list(output_node_names))
/external/tensorflow/tensorflow/contrib/model_pruning/
DREADME.md129 …pruning/strip_pruning_vars --checkpoint_dir=/path/to/checkpoints/ --output_node_names=graph_node1,…
166 …l_pruning/strip_pruning_vars --checkpoint_path=/tmp/cifar10_train --output_node_names=softmax_line…
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dquantization_mnist_test.py136 sess, sess.graph_def, output_node_names=[OUTPUT_NODE_NAME])
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dget_started.md121 --output_node_names=MobileNetV1/Predictions/Reshape_1
128 `output_node_names` may not be obvious outside of the code that built the model.
/external/tensorflow/tensorflow/contrib/quantize/
DREADME.md90 --output_graph=frozen_eval_graph.pb --output_node_names=outputs
/external/tensorflow/tensorflow/compiler/aot/
Dtfcompile.bzl136 " --output_node_names=$$(<$(location " + out_nodes_file +