| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_rendezvous_mgr.cc | 41 GdrRecvTensorCall(WorkerInterface* wi, Device* dst_device, in GdrRecvTensorCall() argument
46 dst_device_(dst_device), in GdrRecvTensorCall()
102 Device* dst_device() const { return dst_device_; } in dst_device() function in tensorflow::__anon6a33ee8b0111::GdrRecvTensorCall
152 Device* dst_device; in RecvFromRemoteAsync() local
153 Status s = sess->device_mgr()->LookupDevice(parsed.dst_device, &dst_device); in RecvFromRemoteAsync()
162 new GdrRecvTensorCall(rwi, dst_device, remote_memory_manager_, in RecvFromRemoteAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker_cache_logger.cc | 89 const string& dst_device, in RecordRecvTensor() argument
92 dst_device, bytes, "", "RecvTensor"); in RecordRecvTensor()
99 const string& dst_device, in RecordDataTransfer() argument
115 " from ", src_device, " to ", dst_device); in RecordDataTransfer()
130 Save(dst_device, step_id, ns); in RecordDataTransfer()
|
| D | collective_rma_distributed_test.cc | 273 Device* dst_device = nullptr; in TEST_F() local
275 TF_EXPECT_OK(device_mgrs_[0]->LookupDevice(dev_name, &dst_device)); in TEST_F()
281 kBufKey, dst_device, to_device_ctx, alloc_attr_, &to_tensor_, in TEST_F()
301 Device* dst_device = nullptr; in TEST_F() local
303 TF_EXPECT_OK(device_mgrs_[0]->LookupDevice(dev_name, &dst_device)); in TEST_F()
309 kBufKey, dst_device, to_device_ctx, alloc_attr_, &to_tensor_, in TEST_F()
333 Device* dst_device = nullptr; in TEST_F() local
335 TF_EXPECT_OK(device_mgrs_[0]->LookupDevice(dev_name, &dst_device)); in TEST_F()
341 kBufKey, dst_device, to_device_ctx, alloc_attr_, &to_tensor_, in TEST_F()
|
| D | base_rendezvous_mgr.cc | 222 if (!is_src && !IsLocalDevice(sess->worker_name, parsed.dst_device)) { in ValidateDevices()
261 Device* dst_device; in SameWorkerRecvDone() local
262 s = sess->device_mgr()->LookupDevice(parsed.dst_device, &dst_device); in SameWorkerRecvDone()
271 Allocator* out_allocator = dst_device->GetAllocator(attr); in SameWorkerRecvDone()
282 recv_args.device_context, src_device, dst_device, in SameWorkerRecvDone()
|
| D | worker_cache_logger.h | 61 const string& dst_device, int64 bytes);
67 const string& dst_device, int64 bytes,
|
| /external/tensorflow/tensorflow/contrib/mpi/ |
| D | mpi_rendezvous_mgr.cc | 75 Device* dst_device; in RecvFromRemoteAsync() local
77 s = env_->device_mgr->LookupDevice(parsed.dst_device, &dst_device); in RecvFromRemoteAsync()
108 Device* dst_device; in RecvFromRemoteAsync() local
110 s = env_->device_mgr->LookupDevice(parsed.dst_device, &dst_device); in RecvFromRemoteAsync()
120 dst_device->MakeTensorFromProto(mpi_response.response().tensor(), in RecvFromRemoteAsync()
124 tr.InitAlloc(dst_device, recv_args.alloc_attrs); in RecvFromRemoteAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr.cc | 62 AllocatorAttributes alloc_attrs, Device* dst_device, in Init() argument
66 dst_device_ = dst_device; in Init()
129 Device* dst_device() const { return dst_device_; } in dst_device() function in tensorflow::__anon2b42d2c80111::RpcRecvTensorCall
240 Device* dst_device; in RecvFromRemoteAsync() local
242 s = sess->device_mgr()->LookupDevice(parsed.dst_device, &dst_device); in RecvFromRemoteAsync()
253 call->Init(rwi, step_id_, parsed.FullKey(), recv_args.alloc_attrs, dst_device, in RecvFromRemoteAsync()
|
| D | grpc_remote_worker.cc | 158 request->dst_device(), num_bytes, "", "RecvBuf"); in RecvBufAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | rendezvous_mgr.cc | 93 Device* dst_device; in SameWorkerRecvDone() local
94 s = device_mgr_->LookupDevice(parsed.dst_device, &dst_device); in SameWorkerRecvDone()
103 Allocator* out_allocator = dst_device->GetAllocator(attr); in SameWorkerRecvDone()
111 recv_args.device_context, src_device, dst_device, in SameWorkerRecvDone()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | rendezvous.cc | 43 dst_device = StringPiece(buf_.data() + (b.dst_device.data() - b_base), in operator =()
44 b.dst_device.size()); in operator =()
53 const string& dst_device, const string& name, in CreateKey() argument
65 dst_device, ";", name, ";", frame_iter.frame_id, ":", frame_iter.iter_id); in CreateKey()
108 out->dst_device = StringPiece(parts[2].data(), parts[2].size()); in ParseKey()
|
| D | rendezvous.h | 57 const string& dst_device, const string& name,
66 StringPiece dst_device; member
|
| D | rendezvous_test.cc | 54 EXPECT_EQ(parsed.dst_device, "/job:mnist/replica:1/task:2/device:GPU:0"); in TEST()
|
| /external/tensorflow/tensorflow/contrib/verbs/ |
| D | rdma_rendezvous_mgr.cc | 55 !DeviceNameUtils::SplitDeviceName(parsed.dst_device, &dst_name, in RecvFromRemoteAsync()
70 s = env_->device_mgr->LookupDevice(parsed.dst_device, &dst_dev); in RecvFromRemoteAsync()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 654 const string& dst_device = dst->assigned_device_name(); in AddControlFlow() local
656 if (src_device == dst_device) continue; in AddControlFlow()
687 const string& cl_key = strings::StrCat(curr_frame_name, "$$", dst_device); in AddControlFlow()
736 const string& dst_device = dst->assigned_device_name(); in AddControlFlow() local
737 if (src_device != dst_device) { in AddControlFlow()
744 strings::StrCat(dst_frame_name, "$$", dst_device); in AddControlFlow()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 468 string dst_device = 9; field
|