Home
last modified time | relevance | path

Searched refs:fetch_tensors (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/cc/client/
H A Dclient_session.cc165 std::vector<Tensor>* fetch_tensors, in RunCallable() argument
167 return impl()->session_->RunCallable(handle, feed_tensors, fetch_tensors, in RunCallable()
173 std::vector<Tensor>* fetch_tensors, in RunCallable() argument
176 return impl()->session_->RunCallable(handle, feed_tensors, fetch_tensors, in RunCallable()
H A Dclient_session.h124 std::vector<Tensor>* fetch_tensors,
137 std::vector<Tensor>* fetch_tensors,
/aosp_15_r20/external/tensorflow/tensorflow/python/client/
H A Dsession_ref.cc329 std::vector<Tensor>* fetch_tensors, in RecordRunCallable() argument
338 RUN_WITH_TIMESTAMP(RunCallable, handle, feed_tensors, fetch_tensors, in RecordRunCallable()
345 for (const Tensor& tensor : *fetch_tensors) { in RecordRunCallable()
472 std::vector<Tensor>* fetch_tensors, in RunCallable() argument
474 LOG_AND_RUN_OPERATION(RunCallable, handle, feed_tensors, fetch_tensors, in RunCallable()
H A Dsession_ref.h72 std::vector<Tensor>* fetch_tensors,
/aosp_15_r20/external/tensorflow/tensorflow/cc/saved_model/
H A Dloader.cc406 std::vector<Tensor>* fetch_tensors, in RunCallable() argument
408 return wrapped_->RunCallable(handle, feed_tensors, fetch_tensors, in RunCallable()
414 std::vector<Tensor>* fetch_tensors, RunMetadata* run_metadata, in RunCallable() argument
416 return wrapped_->RunCallable(handle, feed_tensors, fetch_tensors, in RunCallable()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Ddirect_session.cc1882 std::vector<Tensor>* fetch_tensors) in RunCallableCallFrame() argument
1886 fetch_tensors_(fetch_tensors) {} in RunCallableCallFrame()
1921 std::vector<Tensor>* fetch_tensors, RunMetadata* run_metadata) { in RunCallable() argument
1922 return RunCallable(handle, feed_tensors, fetch_tensors, run_metadata, in RunCallable()
1928 std::vector<Tensor>* fetch_tensors, RunMetadata* run_metadata, in RunCallable() argument
1963 if (fetch_tensors != nullptr) { in RunCallable()
1964 fetch_tensors->resize(executors_and_keys->output_types.size()); in RunCallable()
2002 actual_feed_tensors, fetch_tensors); in RunCallable()
2012 if (fetch_tensors != nullptr) { in RunCallable()
2014 for (auto& tensor : *fetch_tensors) { in RunCallable()
H A Ddirect_session.h126 std::vector<Tensor>* fetch_tensors,
131 std::vector<Tensor>* fetch_tensors, RunMetadata* run_metadata,
H A Ddirect_session_test.cc2904 std::vector<Tensor> fetch_tensors; in TEST() local
2905 TF_ASSERT_OK(session->Run({}, fetch_tensor_names, {}, &fetch_tensors)); in TEST()
2906 ASSERT_EQ(num_outputs_required, fetch_tensors.size()); in TEST()
2907 for (const Tensor& t : fetch_tensors) { in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/core/public/
H A Dsession.h255 std::vector<Tensor>* fetch_tensors, in RunCallable() argument
271 std::vector<Tensor>* fetch_tensors, RunMetadata* run_metadata, in RunCallable() argument
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
H A Dgrpc_session.cc443 std::vector<Tensor>* fetch_tensors, in RunCallable() argument
463 fetch_tensors->push_back(std::move(fetch_tensor)); in RunCallable()
H A Dgrpc_session.h106 std::vector<Tensor>* fetch_tensors,