| /external/tensorflow/tensorflow/core/kernels/ |
| D | sendrecv_ops.cc | 26 static string GetRendezvousKeyPrefix(const string& send_device, in GetRendezvousKeyPrefix() argument
30 return strings::StrCat(send_device, ";", in GetRendezvousKeyPrefix()
56 string send_device; in SendOp() local
57 OP_REQUIRES_OK(ctx, ctx->GetAttr("send_device", &send_device)); in SendOp()
66 key_prefix_ = GetRendezvousKeyPrefix(send_device, recv_device, in SendOp()
124 string send_device; in RecvOp() local
125 OP_REQUIRES_OK(ctx, ctx->GetAttr("send_device", &send_device)); in RecvOp()
134 key_prefix_ = GetRendezvousKeyPrefix(send_device, recv_device, in RecvOp()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | cluster_function_library_runtime_test.cc | 131 key: "send_device" in TEST_F()
172 key: "send_device" in TEST_F()
239 key: "send_device" in TEST_F()
281 key: "send_device" in TEST_F()
|
| D | master_session.cc | 398 string send_device; in TrackFeedsAndFetches() local
399 TF_CHECK_OK(GetNodeAttr(ndef, "send_device", &send_device)); in TrackFeedsAndFetches()
407 Rendezvous::CreateKey(send_device, send_device_incarnation, in TrackFeedsAndFetches()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | kernel_benchmark_testlib.cc | 118 string send_device; in GetRendezvousKey() local
119 TF_CHECK_OK(GetNodeAttr(node->attrs(), "send_device", &send_device)); in GetRendezvousKey()
127 return Rendezvous::CreateKey(send_device, send_device_incarnation, in GetRendezvousKey()
|
| D | step_stats_collector.cc | 90 string send_device; in Done() local
91 TF_CHECK_OK(GetNodeAttr(attrs, "send_device", &send_device)); in Done()
93 "(", tensor_name, " @", send_device); in Done()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_group_metadata.cc | 452 auto send_device = GetInstructionDevice(*channel.send); in VerifyChannelInstructions() local
454 if (!send_device) { in VerifyChannelInstructions()
462 if (*send_device != *send_done_device) { in VerifyChannelInstructions()
466 channel.id, *send_device, *send_done_device); in VerifyChannelInstructions()
480 if (*send_device == *recv_device) { in VerifyChannelInstructions()
483 channel.id, *send_device); in VerifyChannelInstructions()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute.cc | 1086 const string& send_device, int64 send_device_incarnation, in ExecuteRecv() argument
1098 op.MutableAttrs()->Set("send_device", send_device); in ExecuteRecv()
1129 tensorflow::Device* send_device = h->device(); in EagerCopyToDevice() local
1131 if (send_device == nullptr) { in EagerCopyToDevice()
1132 send_device = ctx->HostCPU(); in EagerCopyToDevice()
1135 bool sender_is_local = IsLocal(ctx, send_device); in EagerCopyToDevice()
1150 ExecuteSend(ctx, send_device, h, wire_id, recv_device->name())); in EagerCopyToDevice()
1152 return ExecuteRecv(ctx, recv_device, h->dtype, wire_id, send_device->name(), in EagerCopyToDevice()
1153 send_device->attributes().incarnation(), result); in EagerCopyToDevice()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental_test.cc | 241 string send_device = "/job:localhost/replica:0/task:0/device:CPU:0"; in TEST() local
242 TFE_OpSetAttrString(send_op, "send_device", send_device.c_str(), in TEST()
243 send_device.size()); in TEST()
257 TFE_OpSetAttrString(recv_op, "send_device", send_device.c_str(), in TEST()
258 send_device.size()); in TEST()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 926 string send_device; in SetIncarnation() local
927 if (!GetNodeAttr(*ndef, "send_device", &send_device).ok()) { in SetIncarnation()
934 incarnation = opts.get_incarnation(send_device); in SetIncarnation()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler_test.cc | 350 key: "send_device" in CreateGrapplerItemWithSendRecv()
385 key: "send_device" in CreateGrapplerItemWithSendRecv()
442 key: "send_device" in CreateGrapplerItemWithRecvWithoutSend()
|