Searched refs:fetch_tensors (Results 1 – 14 of 14) sorted by relevance
/external/tensorflow/tensorflow/contrib/predictor/ |
D | predictor.py | 42 def fetch_tensors(self): member in Predictor 77 return self._session.run(fetches=self.fetch_tensors, feed_dict=feed_dict)
|
D | contrib_estimator_predictor.py | 77 _, fetch_tensors = output_alternatives[output_alternative_key] 78 self._fetch_tensors = fetch_tensors
|
D | saved_model_predictor_test.py | 70 output_tensor_name = predictor.fetch_tensors['outputs'].name 113 output_tensor_name = predictor.fetch_tensors[key].name 140 output_tensor_name = predictor.fetch_tensors[key].name
|
D | contrib_estimator_predictor_test.py | 57 output_tensor_name = predictor.fetch_tensors[key].name
|
D | core_estimator_predictor_test.py | 68 output_tensor_name = predictor.fetch_tensors[output_key].name
|
/external/tensorflow/tensorflow/python/client/ |
D | session_ref.cc | 327 std::vector<Tensor>* fetch_tensors, in RecordRunCallable() argument 336 RUN_WITH_TIMESTAMP(RunCallable, handle, feed_tensors, fetch_tensors, in RecordRunCallable() 343 for (const Tensor& tensor : *fetch_tensors) { in RecordRunCallable() 470 std::vector<Tensor>* fetch_tensors, in RunCallable() argument 472 LOG_AND_RUN_OPERATION(RunCallable, handle, feed_tensors, fetch_tensors, in RunCallable()
|
D | session_ref.h | 72 std::vector<Tensor>* fetch_tensors,
|
/external/tensorflow/tensorflow/cc/client/ |
D | client_session.cc | 138 std::vector<Tensor>* fetch_tensors, in RunCallable() argument 140 return impl()->session_->RunCallable(handle, feed_tensors, fetch_tensors, in RunCallable()
|
D | client_session.h | 110 std::vector<Tensor>* fetch_tensors,
|
/external/tensorflow/tensorflow/core/public/ |
D | session.h | 220 std::vector<Tensor>* fetch_tensors, in RunCallable() argument
|
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
D | grpc_session.h | 102 std::vector<Tensor>* fetch_tensors,
|
D | grpc_session.cc | 424 std::vector<Tensor>* fetch_tensors, in RunCallable() argument 444 fetch_tensors->push_back(std::move(fetch_tensor)); in RunCallable()
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | direct_session.cc | 1744 std::vector<Tensor>* fetch_tensors) in RunCallableCallFrame() argument 1748 fetch_tensors_(fetch_tensors) {} in RunCallableCallFrame() 1786 std::vector<Tensor>* fetch_tensors, RunMetadata* run_metadata) { in RunCallable() argument 1820 if (fetch_tensors != nullptr) { in RunCallable() 1821 fetch_tensors->resize(executors_and_keys->output_types.size()); in RunCallable() 1832 fetch_tensors); in RunCallable()
|
D | direct_session.h | 115 std::vector<Tensor>* fetch_tensors,
|