| /external/tensorflow/tensorflow/core/kernels/ |
| D | sendrecv_ops.cc | 27 const string& recv_device, in GetRendezvousKeyPrefix() argument
32 recv_device, ";", tensor_name); in GetRendezvousKeyPrefix()
58 string recv_device; in SendOp() local
59 OP_REQUIRES_OK(ctx, ctx->GetAttr("recv_device", &recv_device)); in SendOp()
66 key_prefix_ = GetRendezvousKeyPrefix(send_device, recv_device, in SendOp()
126 string recv_device; in RecvOp() local
127 OP_REQUIRES_OK(ctx, ctx->GetAttr("recv_device", &recv_device)); in RecvOp()
134 key_prefix_ = GetRendezvousKeyPrefix(send_device, recv_device, in RecvOp()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute.cc | 576 Device* recv_device, TensorHandle** result) { in EagerRemoteSendTensor() argument
584 ctx->GetClientAndContextID(recv_device, &eager_client, &context_id)); in EagerRemoteSendTensor()
591 request.set_device_name(recv_device->name()); in EagerRemoteSendTensor()
630 /*d=*/recv_device, /*op_device=*/recv_device, in EagerRemoteSendTensor()
1056 const string& recv_device) { in ExecuteSend() argument
1073 op.MutableAttrs()->Set("recv_device", recv_device); in ExecuteSend()
1137 tensorflow::Device* recv_device; in EagerCopyToDevice() local
1138 TF_RETURN_IF_ERROR(FindDeviceFromName(ctx, device_name, &recv_device)); in EagerCopyToDevice()
1140 bool recver_is_local = IsLocal(ctx, recv_device); in EagerCopyToDevice()
1143 return LocalEagerCopyToDevice(h, ctx, recv_device, result); in EagerCopyToDevice()
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | cluster_function_library_runtime_test.cc | 125 key: "recv_device" in TEST_F()
166 key: "recv_device" in TEST_F()
233 key: "recv_device" in TEST_F()
275 key: "recv_device" in TEST_F()
|
| D | master_session.cc | 400 string recv_device; in TrackFeedsAndFetches() local
401 TF_CHECK_OK(GetNodeAttr(ndef, "recv_device", &recv_device)); in TrackFeedsAndFetches()
408 recv_device, name, FrameAndIter(0, 0)); in TrackFeedsAndFetches()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | kernel_benchmark_testlib.cc | 120 string recv_device; in GetRendezvousKey() local
121 TF_CHECK_OK(GetNodeAttr(node->attrs(), "recv_device", &recv_device)); in GetRendezvousKey()
128 recv_device, tensor_name, FrameAndIter(0, 0)); in GetRendezvousKey()
|
| D | step_stats_collector.cc | 83 string recv_device; in Done() local
84 TF_CHECK_OK(GetNodeAttr(attrs, "recv_device", &recv_device)); in Done()
86 "(", tensor_name, " @", recv_device); in Done()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental_test.cc | 245 string recv_device = "/job:localhost/replica:0/task:0/device:CPU:0"; in TEST() local
246 TFE_OpSetAttrString(send_op, "recv_device", recv_device.c_str(), in TEST()
247 recv_device.size()); in TEST()
260 TFE_OpSetAttrString(recv_op, "recv_device", recv_device.c_str(), in TEST()
261 recv_device.size()); in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_group_metadata.cc | 468 auto recv_device = GetInstructionDevice(*channel.recv); in VerifyChannelInstructions() local
474 if (*recv_device != *recv_done_device) { in VerifyChannelInstructions()
478 channel.id, *recv_device, *recv_done_device); in VerifyChannelInstructions()
480 if (*send_device == *recv_device) { in VerifyChannelInstructions()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler_test.cc | 344 key: "recv_device" in CreateGrapplerItemWithSendRecv()
379 key: "recv_device" in CreateGrapplerItemWithSendRecv()
436 key: "recv_device" in CreateGrapplerItemWithRecvWithoutSend()
2857 const string recv_device = kCPU1; in TEST_F() local
2861 node->set_device(recv_device); in TEST_F()
2863 (*attr)["recv_device"].set_s(recv_device); in TEST_F()
2866 (*attr)["recv_device"].set_s(recv_device); in TEST_F()
|