Home
last modified time | relevance | path

Searched refs:output_tensor_names (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/contrib/session_bundle/
Dsignature.cc118 std::vector<string> output_tensor_names; in RunClassification() local
120 output_tensor_names.push_back(signature.classes().tensor_name()); in RunClassification()
123 output_tensor_names.push_back(signature.scores().tensor_name()); in RunClassification()
129 output_tensor_names, {}, &outputs); in RunClassification()
136 if (outputs.size() != output_tensor_names.size()) { in RunClassification()
138 strings::StrCat("Expected ", output_tensor_names.size(), in RunClassification()
173 std::vector<string> output_tensor_names; in RunRegression() local
175 output_tensor_names.push_back(signature.output().tensor_name()); in RunRegression()
181 output_tensor_names, {}, &outputs); in RunRegression()
188 if (outputs.size() != output_tensor_names.size()) { in RunRegression()
[all …]
Dsignature_test.cc255 output_tensor_names = output_tensor_names_arg; in Run()
271 std::vector<string> output_tensor_names; member
318 ASSERT_EQ(2, session_.output_tensor_names.size()); in TEST_F()
319 EXPECT_EQ(kClassesName, session_.output_tensor_names[0]); in TEST_F()
320 EXPECT_EQ(kScoresName, session_.output_tensor_names[1]); in TEST_F()
339 ASSERT_EQ(1, session_.output_tensor_names.size()); in TEST_F()
340 EXPECT_EQ(kClassesName, session_.output_tensor_names[0]); in TEST_F()
359 ASSERT_EQ(1, session_.output_tensor_names.size()); in TEST_F()
360 EXPECT_EQ(kScoresName, session_.output_tensor_names[0]); in TEST_F()
445 ASSERT_EQ(1, session_.output_tensor_names.size()); in TEST_F()
[all …]
DREADME.md191 `output_tensor_names` in the `tensorflow::Session->Run()` arguments. The next
194 supplied `output_tensor_names`.
/external/tensorflow/tensorflow/contrib/model_pruning/python/
Dstrip_pruning_vars_test.py183 output_tensor_names = [top_layer.name]
184 initial_outputs = self._get_initial_outputs(output_tensor_names)
188 final_outputs = self._get_final_outputs(output_tensor_names)
200 output_tensor_names = [top_layer.name]
201 initial_outputs = self._get_initial_outputs(output_tensor_names)
205 final_outputs = self._get_final_outputs(output_tensor_names)
217 output_tensor_names = [outputs[0][0].name]
218 initial_outputs = self._get_initial_outputs(output_tensor_names)
222 final_outputs = self._get_final_outputs(output_tensor_names)
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_session.cc185 const std::vector<string>& output_tensor_names, in RunHelper() argument
215 for (int i = 0; i < output_tensor_names.size(); ++i) { in RunHelper()
216 const string& name = output_tensor_names[i]; in RunHelper()
234 if (!output_tensor_names.empty()) { in RunHelper()
235 outputs->resize(output_tensor_names.size()); in RunHelper()
252 if (output_name_to_offset.size() != output_tensor_names.size()) { in RunHelper()
253 for (int i = 0; i < output_tensor_names.size(); ++i) { in RunHelper()
254 const string& name = output_tensor_names[i]; in RunHelper()
271 const std::vector<string>& output_tensor_names, in Run() argument
275 return RunHelper(run_options, inputs, output_tensor_names, target_node_names, in Run()
[all …]
Dgrpc_session.h70 const std::vector<string>& output_tensor_names,
75 const std::vector<string>& output_tensor_names,
128 const std::vector<string>& output_tensor_names,
/external/tensorflow/tensorflow/python/client/
Dsession_ref.cc120 const std::vector<string>& output_tensor_names, in RecordRun() argument
123 return RecordRun(session, *kEmptyRunOptions(), inputs, output_tensor_names, in RecordRun()
129 const std::vector<string>& output_tensor_names, in RecordRun() argument
148 for (int i = 0; i < output_tensor_names.size(); ++i) { in RecordRun()
149 const string& name = output_tensor_names[i]; in RecordRun()
159 RUN_WITH_TIMESTAMP(Run, inputs, output_tensor_names, target_node_names, in RecordRun()
162 RUN_WITH_TIMESTAMP(Run, run_options, inputs, output_tensor_names, in RecordRun()
170 tproto->set_name(output_tensor_names[i]); in RecordRun()
410 const std::vector<string>& output_tensor_names, in Run() argument
414 LOG_AND_RUN_OPERATION(Run, run_options, inputs, output_tensor_names, in Run()
[all …]
Dsession_ref.h42 const std::vector<string>& output_tensor_names,
53 const std::vector<string>& output_tensor_names,
/external/tensorflow/tensorflow/cc/client/
Dclient_session.cc115 std::vector<string> output_tensor_names; in Run() local
116 output_tensor_names.reserve(fetch_outputs.size()); in Run()
118 output_tensor_names.push_back(output.name()); in Run()
126 return impl()->session_->Run(run_options, feeds, output_tensor_names, in Run()
/external/tensorflow/tensorflow/core/public/
Dsession.h122 const std::vector<string>& output_tensor_names,
152 const std::vector<string>& output_tensor_names,
/external/tensorflow/tensorflow/tools/benchmark/
Dbenchmark_model.cc122 std::vector<string> output_tensor_names; in GetOutputShapes() local
134 output_tensor_names.push_back(wanted_shape); in GetOutputShapes()
138 session->Run(input_tensors, output_tensor_names, {}, &output_tensors)); in GetOutputShapes()
139 CHECK_EQ(output_tensors.size(), output_tensor_names.size()); in GetOutputShapes()
140 for (int i = 0; i < output_tensor_names.size(); ++i) { in GetOutputShapes()
141 const string& wanted_shape_name = output_tensor_names[i]; in GetOutputShapes()
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
Dcandidate_graph_runner.cc71 const std::vector<string>& output_tensor_names, in RunOp() argument
77 TF_CHECK_OK(session_->Run(inputs, output_tensor_names, op_name, outputs)) in RunOp()
Dcandidate_graph_runner.h62 const std::vector<string>& output_tensor_names,
/external/tensorflow/tensorflow/python/framework/
Dfunction_def_to_graph.py69 output_tensor_names = [
74 func_graph.get_tensor_by_name(name) for name in output_tensor_names
/external/tensorflow/tensorflow/core/common_runtime/
Dsession.cc31 const std::vector<string>& output_tensor_names, in Run() argument
/external/tensorflow/tensorflow/cc/saved_model/
Dloader.cc114 const std::vector<string>& output_tensor_names, in RunOnce() argument
127 for (const string& output_tensor_name : output_tensor_names) { in RunOnce()
/external/tensorflow/tensorflow/c/
Dc_api.cc838 const std::vector<string>& output_tensor_names, TF_Tensor** c_outputs, in TF_Run_Helper() argument
842 const int noutputs = output_tensor_names.size(); in TF_Run_Helper()
860 result = session->Run(run_options_proto, input_pairs, output_tensor_names, in TF_Run_Helper()
870 result = session->PRun(handle, input_pairs, output_tensor_names, &outputs); in TF_Run_Helper()