| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker_cache_logger.cc | 88 const string& src_device, in RecordRecvTensor() argument
91 RecordDataTransfer(step_id, start_usecs, end_usecs, tensor_name, src_device, in RecordRecvTensor()
98 const string& src_device, in RecordDataTransfer() argument
115 " from ", src_device, " to ", dst_device); in RecordDataTransfer()
|
| D | base_rendezvous_mgr.cc | 195 if (!IsLocalDevice(session_->worker_name, parsed.src_device)) { in Send()
218 if (is_src && !IsLocalDevice(sess->worker_name, parsed.src_device)) { in ValidateDevices()
255 Device* src_device; in SameWorkerRecvDone() local
256 Status s = sess->device_mgr()->LookupDevice(parsed.src_device, &src_device); in SameWorkerRecvDone()
282 recv_args.device_context, src_device, dst_device, in SameWorkerRecvDone()
|
| D | worker_cache_logger.h | 60 const string& tensor_name, const string& src_device,
66 const string& tensor_name, const string& src_device,
|
| D | worker.cc | 434 string local_name = DeviceNameUtils::LocalName(parsed.src_device); in PrepareRecvTensor()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | rendezvous.cc | 39 src_device = StringPiece(buf_.data() + (b.src_device.data() - b_base), in operator =()
40 b.src_device.size()); in operator =()
52 string Rendezvous::CreateKey(const string& src_device, uint64 src_incarnation, in CreateKey() argument
64 src_device, ";", strings::Uint64ToHexString(src_incarnation, buf), ";", in CreateKey()
107 out->src_device = StringPiece(parts[0].data(), parts[0].size()); in ParseKey()
|
| D | rendezvous.h | 56 static string CreateKey(const string& src_device, uint64 src_incarnation,
63 StringPiece src_device; member
|
| D | rendezvous_test.cc | 51 EXPECT_EQ(parsed.src_device, "/job:mnist/replica:1/task:2/CPU:0"); in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | rendezvous_mgr.cc | 87 Device* src_device; in SameWorkerRecvDone() local
88 Status s = device_mgr_->LookupDevice(parsed.src_device, &src_device); in SameWorkerRecvDone()
111 recv_args.device_context, src_device, dst_device, in SameWorkerRecvDone()
|
| D | process_function_library_runtime.cc | 331 const string* src_device = AssignedOrRequestedDeviceName(*src_node); in PinArgsAndRets() local
335 << " src_device: " << *src_device in PinArgsAndRets()
337 while (src_device->empty() && colocation_group.empty() && in PinArgsAndRets()
340 src_device = AssignedOrRequestedDeviceName(*src_node); in PinArgsAndRets()
343 << " src_device: " << *src_device in PinArgsAndRets()
352 } else if (!src_device->empty()) { in PinArgsAndRets()
356 if (!DeviceNameUtils::ParseFullName(*src_device, &parsed)) { in PinArgsAndRets()
359 *src_device); in PinArgsAndRets()
365 "Unable to find any devices for spec ", *src_device); in PinArgsAndRets()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_rendezvous_mgr.cc | 136 if (!DeviceNameUtils::SplitDeviceName(parsed.src_device, &src_worker, in RecvFromRemoteAsync()
138 Status s = errors::Internal(parsed.src_device, in RecvFromRemoteAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr.cc | 226 if (!DeviceNameUtils::SplitDeviceName(parsed.src_device, &call->src_worker_, in RecvFromRemoteAsync()
228 s = errors::Internal(parsed.src_device, in RecvFromRemoteAsync()
|
| D | grpc_remote_worker.cc | 157 step_id, send_start_usec, end_usec, key, request->src_device(), in RecvBufAsync()
|
| /external/tensorflow/tensorflow/contrib/verbs/ |
| D | rdma_rendezvous_mgr.cc | 53 if (!DeviceNameUtils::SplitDeviceName(parsed.src_device, &src_name, in RecvFromRemoteAsync()
|
| D | rdma.cc | 1021 string local_name = DeviceNameUtils::LocalName(parsed.src_device); in PrepareRecvTensor()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 653 const string& src_device = src->assigned_device_name(); in AddControlFlow() local
656 if (src_device == dst_device) continue; in AddControlFlow()
735 const string& src_device = src->assigned_device_name(); in AddControlFlow() local
737 if (src_device != dst_device) { in AddControlFlow()
|
| /external/tensorflow/tensorflow/contrib/mpi/ |
| D | mpi_rendezvous_mgr.cc | 218 parsed.src_device, &src_dev); in AddRequest()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 467 string src_device = 8; field
|