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.cc554 ProtoRunGraphRequest::ProtoRunGraphRequest(const RunGraphRequest* request) in ProtoRunGraphRequest() function in tensorflow::ProtoRunGraphRequest
557 const string& ProtoRunGraphRequest::session_handle() const { in session_handle()
561 bool ProtoRunGraphRequest::create_worker_session_called() const { in create_worker_session_called()
565 const string& ProtoRunGraphRequest::graph_handle() const { in graph_handle()
569 int64 ProtoRunGraphRequest::step_id() const { return request_->step_id(); } in step_id()
571 const ExecutorOpts& ProtoRunGraphRequest::exec_opts() const { in exec_opts()
575 size_t ProtoRunGraphRequest::num_sends() const { return request_->send_size(); } in num_sends()
577 const string& ProtoRunGraphRequest::send_key(size_t i) const { in send_key()
581 Status ProtoRunGraphRequest::SendValue(size_t i, Tensor* out_tensor) const { in SendValue()
589 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.h66 RunGraphRequestWrapper* wrapped_request = new ProtoRunGraphRequest(request); in RunGraphAsync()
Dmessage_wrappers.h432 class ProtoRunGraphRequest : public RunGraphRequestWrapper {
434 ProtoRunGraphRequest(const RunGraphRequest* request);
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_worker_service.cc244 ProtoRunGraphRequest* wrapped_request = in RunGraphHandler()
245 new ProtoRunGraphRequest(&call->request); in RunGraphHandler()