Home
last modified time | relevance | path

Searched refs:feed_tensors (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/contrib/predictor/
Dpredictor.py38 def feed_tensors(self): member in Predictor
65 expected_keys = set(self.feed_tensors.keys())
73 for key in self.feed_tensors.keys():
76 feed_dict[self.feed_tensors[key]] = value
/external/tensorflow/tensorflow/python/client/
Dsession_ref.cc326 const std::vector<Tensor>& feed_tensors, in RecordRunCallable() argument
333 for (auto& tensor : feed_tensors) { in RecordRunCallable()
336 RUN_WITH_TIMESTAMP(RunCallable, handle, feed_tensors, fetch_tensors, in RecordRunCallable()
469 const std::vector<Tensor>& feed_tensors, in RunCallable() argument
472 LOG_AND_RUN_OPERATION(RunCallable, handle, feed_tensors, fetch_tensors, in RunCallable()
Dsession_ref.h71 const std::vector<Tensor>& feed_tensors,
/external/tensorflow/tensorflow/cc/client/
Dclient_session.cc137 const std::vector<Tensor>& feed_tensors, in RunCallable() argument
140 return impl()->session_->RunCallable(handle, feed_tensors, fetch_tensors, in RunCallable()
Dclient_session.h109 const std::vector<Tensor>& feed_tensors,
/external/tensorflow/tensorflow/cc/saved_model/
Dloader.cc119 std::vector<Tensor> feed_tensors; in RunOnce() local
125 feed_tensors.push_back(tensor); in RunOnce()
136 const Status run_status = session->RunCallable(callable_handle, feed_tensors, in RunOnce()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.cc437 std::set<std::pair<string, int>> feed_tensors; in PruneGraphDefInto() local
439 feed_tensors.insert( in PruneGraphDefInto()
473 if (feed_tensors.find(std::make_pair(node_name, id.second)) == in PruneGraphDefInto()
474 feed_tensors.end()) { in PruneGraphDefInto()
/external/tensorflow/tensorflow/core/public/
Dsession.h219 const std::vector<Tensor>& feed_tensors, in RunCallable() argument
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_session.h101 const std::vector<Tensor>& feed_tensors,
Dgrpc_session.cc423 const std::vector<Tensor>& feed_tensors, in RunCallable() argument
430 for (const Tensor& feed : feed_tensors) { in RunCallable()
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session.cc1743 const std::vector<Tensor>* feed_tensors, in RunCallableCallFrame() argument
1747 feed_tensors_(feed_tensors), in RunCallableCallFrame()
1785 CallableHandle handle, const std::vector<Tensor>& feed_tensors, in RunCallable() argument
1815 if (feed_tensors.size() != executors_and_keys->input_types.size()) { in RunCallable()
1818 " feed tensors, but got ", feed_tensors.size()); in RunCallable()
1831 RunCallableCallFrame call_frame(this, executors_and_keys.get(), &feed_tensors, in RunCallable()
Ddirect_session.h114 const std::vector<Tensor>& feed_tensors,