Home
last modified time | relevance | path

Searched refs:ProtoRunGraphRequest (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/
Dmessage_wrappers.cc539 ProtoRunGraphRequest::ProtoRunGraphRequest(const RunGraphRequest* request) in ProtoRunGraphRequest() function in tensorflow::ProtoRunGraphRequest
542 const string& ProtoRunGraphRequest::session_handle() const { in session_handle()
546 bool ProtoRunGraphRequest::create_worker_session_called() const { in create_worker_session_called()
550 const string& ProtoRunGraphRequest::graph_handle() const { in graph_handle()
554 int64 ProtoRunGraphRequest::step_id() const { return request_->step_id(); } in step_id()
556 const ExecutorOpts& ProtoRunGraphRequest::exec_opts() const { in exec_opts()
560 size_t ProtoRunGraphRequest::num_sends() const { return request_->send_size(); } in num_sends()
562 const string& ProtoRunGraphRequest::send_key(size_t i) const { in send_key()
566 Status ProtoRunGraphRequest::SendValue(size_t i, Tensor* out_tensor) const { in SendValue()
574 size_t ProtoRunGraphRequest::num_recvs() const { in num_recvs()
[all …]
Dmessage_wrappers_test.cc195 CheckRunGraphRequest(ProtoRunGraphRequest(&request.ToProto())); in TEST()
203 CheckRunGraphRequest(ProtoRunGraphRequest(&request.ToProto())); in TEST()
211 CheckRunGraphRequest(ProtoRunGraphRequest(&request.ToProto())); in TEST()
219 CheckRunGraphRequest(ProtoRunGraphRequest(&request.ToProto())); in TEST()
227 CheckRunGraphRequest(ProtoRunGraphRequest(&request.ToProto())); in TEST()
235 CheckRunGraphRequest(ProtoRunGraphRequest(&request.ToProto())); in TEST()
Dworker_interface.h67 RunGraphRequestWrapper* wrapped_request = new ProtoRunGraphRequest(request); in RunGraphAsync()
Dmessage_wrappers.h421 class ProtoRunGraphRequest : public RunGraphRequestWrapper {
423 ProtoRunGraphRequest(const RunGraphRequest* request);
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_worker_service.cc227 ProtoRunGraphRequest* wrapped_request = in RunGraphHandler()
228 new ProtoRunGraphRequest(&call->request); in RunGraphHandler()