| /external/tensorflow/tensorflow/cc/client/ |
| D | client_session.cc | 115 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()
141 std::vector<string> output_tensor_names; in Run() local
142 output_tensor_names.reserve(fetch_outputs.size()); in Run()
144 output_tensor_names.push_back(output.name()); in Run()
152 return impl()->session_->Run(run_options, feeds, output_tensor_names, in Run()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_session.cc | 202 const std::vector<string>& output_tensor_names, in RunHelper() argument
232 for (int i = 0, end = output_tensor_names.size(); i < end; ++i) { in RunHelper()
233 const string& name = output_tensor_names[i]; in RunHelper()
251 if (!output_tensor_names.empty()) { in RunHelper()
252 outputs->resize(output_tensor_names.size()); in RunHelper()
269 if (output_name_to_offset.size() != output_tensor_names.size()) { in RunHelper()
270 for (int i = 0, end = output_tensor_names.size(); i < end; ++i) { in RunHelper()
271 const string& name = output_tensor_names[i]; in RunHelper()
288 const std::vector<string>& output_tensor_names, in Run() argument
292 return RunHelper(run_options, inputs, output_tensor_names, target_node_names, in Run()
[all …]
|
| D | grpc_session.h | 72 const std::vector<string>& output_tensor_names,
77 const std::vector<string>& output_tensor_names,
134 const std::vector<string>& output_tensor_names,
|
| /external/tensorflow/tensorflow/python/client/ |
| D | session_ref.cc | 121 const std::vector<string>& output_tensor_names, in RecordRun() argument
124 return RecordRun(session, *kEmptyRunOptions(), inputs, output_tensor_names, in RecordRun()
130 const std::vector<string>& output_tensor_names, in RecordRun() argument
149 for (int i = 0, end = output_tensor_names.size(); i < end; ++i) { in RecordRun()
150 const string& name = output_tensor_names[i]; in RecordRun()
160 RUN_WITH_TIMESTAMP(Run, inputs, output_tensor_names, target_node_names, in RecordRun()
163 RUN_WITH_TIMESTAMP(Run, run_options, inputs, output_tensor_names, in RecordRun()
171 tproto->set_name(output_tensor_names[i]); in RecordRun()
411 const std::vector<string>& output_tensor_names, in Run() argument
415 LOG_AND_RUN_OPERATION(Run, run_options, inputs, output_tensor_names, in Run()
[all …]
|
| D | session_ref.h | 42 const std::vector<string>& output_tensor_names,
53 const std::vector<string>& output_tensor_names,
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/ |
| D | model_handler.py | 230 def output_tensor_names(self) -> Sequence[str]: member in _ModelHandlerBase
286 def output_tensor_names(self) -> Sequence[str]: member in ModelHandlerV1
312 output_tensor_names = self.output_tensor_names
314 sess.run(fetches=output_tensor_names, feed_dict=inputs)
318 outputs = sess.run(fetches=output_tensor_names, feed_dict=inputs)
327 output_names=self.output_tensor_names,
347 def output_tensor_names(self): member in ModelHandlerV2
382 output_names=self.output_tensor_names,
452 conversion_nodes_denylist = self.output_tensor_names
484 fetch_names=self.output_tensor_names,
|
| /external/tensorflow/tensorflow/core/public/ |
| D | session.h | 135 const std::vector<std::string>& output_tensor_names,
173 const std::vector<std::string>& output_tensor_names,
182 const std::vector<std::string>& output_tensor_names, in Run() argument
|
| /external/tensorflow/tensorflow/cc/saved_model/ |
| D | loader.cc | 153 const std::vector<string>& output_tensor_names, in RunOnce() argument
166 for (const string& output_tensor_name : output_tensor_names) { in RunOnce()
319 const std::vector<string>& output_tensor_names, in Run() argument
322 return wrapped_->Run(inputs, output_tensor_names, target_node_names, in Run()
344 const std::vector<string>& output_tensor_names, in Run() argument
347 return wrapped_->Run(run_options, inputs, output_tensor_names, in Run()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | function_def_to_graph.py | 81 output_tensor_names = [
86 func_graph.get_tensor_by_name(name) for name in output_tensor_names
107 fdef.signature.output_arg, output_tensor_names):
|
| /external/tensorflow/tensorflow/tools/benchmark/ |
| D | benchmark_model.cc | 123 std::vector<string> output_tensor_names; in GetOutputShapes() local
135 output_tensor_names.push_back(wanted_shape); in GetOutputShapes()
139 session->Run(input_tensors, output_tensor_names, {}, &output_tensors)); in GetOutputShapes()
140 CHECK_EQ(output_tensors.size(), output_tensor_names.size()); in GetOutputShapes()
141 for (int i = 0; i < output_tensor_names.size(); ++i) { in GetOutputShapes()
142 const string& wanted_shape_name = output_tensor_names[i]; in GetOutputShapes()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | session.cc | 39 const std::vector<string>& output_tensor_names, in Run() argument
|
| /external/tflite-support/tensorflow_lite_support/codegen/ |
| D | android_java_generator.cc | 183 std::vector<std::string> output_tensor_names = std::move(names.second); in CreateModelInfo() local
197 for (int i = 0; i < output_tensor_names.size(); i++) { in CreateModelInfo()
200 output_tensor_names[i], false, i, err)); in CreateModelInfo()
201 if (i < output_tensor_names.size() - 1) { in CreateModelInfo()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 414 const std::vector<string>& output_tensor_names, TF_Tensor** c_outputs, in TF_Run_Helper() argument
418 const int noutputs = output_tensor_names.size(); in TF_Run_Helper()
436 result = session->Run(run_options_proto, input_pairs, output_tensor_names, in TF_Run_Helper()
446 result = session->PRun(handle, input_pairs, output_tensor_names, &outputs); in TF_Run_Helper()
|