| /external/tensorflow/tensorflow/core/kernels/ |
| sendrecv_ops.cc | 27 const string& recv_device,
32 recv_device, ";", tensor_name);
58 string recv_device; local
59 OP_REQUIRES_OK(ctx, ctx->GetAttr("recv_device", &recv_device));
66 key_prefix_ = GetRendezvousKeyPrefix(send_device, recv_device,
126 string recv_device; local
127 OP_REQUIRES_OK(ctx, ctx->GetAttr("recv_device", &recv_device));
134 key_prefix_ = GetRendezvousKeyPrefix(send_device, recv_device,
[all...] |
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| cluster_function_library_runtime_test.cc | 125 key: "recv_device"
166 key: "recv_device"
233 key: "recv_device"
275 key: "recv_device"
|
| master_session.cc | 400 string recv_device; local
401 TF_CHECK_OK(GetNodeAttr(ndef, "recv_device", &recv_device));
408 recv_device, name, FrameAndIter(0, 0));
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| kernel_benchmark_testlib.cc | 120 string recv_device; local
121 TF_CHECK_OK(GetNodeAttr(node->attrs(), "recv_device", &recv_device));
128 recv_device, tensor_name, FrameAndIter(0, 0));
|
| step_stats_collector.cc | 83 string recv_device; local
84 TF_CHECK_OK(GetNodeAttr(attrs, "recv_device", &recv_device));
86 "(", tensor_name, " @", recv_device);
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| execute.cc | 576 Device* recv_device, TensorHandle** result) {
584 ctx->GetClientAndContextID(recv_device, &eager_client, &context_id));
591 request.set_device_name(recv_device->name());
630 /*d=*/recv_device, /*op_device=*/recv_device,
[all...] |
| /external/tensorflow/tensorflow/c/ |
| c_api_experimental_test.cc | 245 string recv_device = "/job:localhost/replica:0/task:0/device:CPU:0"; local
246 TFE_OpSetAttrString(send_op, "recv_device", recv_device.c_str(),
247 recv_device.size());
260 TFE_OpSetAttrString(recv_op, "recv_device", recv_device.c_str(),
261 recv_device.size());
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_module_group_metadata.cc | 468 auto recv_device = GetInstructionDevice(*channel.recv); local
474 if (*recv_device != *recv_done_device) {
478 channel.id, *recv_device, *recv_done_device);
480 if (*send_device == *recv_device) {
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| virtual_scheduler_test.cc | 344 key: "recv_device"
379 key: "recv_device"
436 key: "recv_device"
2857 const string recv_device = kCPU1; local
[all...] |
