| /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()
121 string send_device; in RecvOp() local
122 OP_REQUIRES_OK(ctx, ctx->GetAttr("send_device", &send_device)); in RecvOp()
131 key_prefix_ = GetRendezvousKeyPrefix(send_device, recv_device, in RecvOp()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_Send.pbtxt | 17 name: "send_device"
25 The current incarnation of send_device.
43 summary: "Sends the named tensor from send_device to recv_device."
|
| D | api_def_Recv.pbtxt | 17 name: "send_device"
25 The current incarnation of send_device.
43 summary: "Receives the named tensor from send_device on recv_device."
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental_test.cc | 149 string send_device = "/job:localhost/replica:0/task:0/device:CPU:0"; in TEST() local
150 TFE_OpSetAttrString(send_op, "send_device", send_device.c_str(), in TEST()
151 send_device.size()); in TEST()
165 TFE_OpSetAttrString(recv_op, "send_device", send_device.c_str(), in TEST()
166 send_device.size()); in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | kernel_benchmark_testlib.cc | 123 string send_device; in GetRendezvousKey() local
124 TF_CHECK_OK(GetNodeAttr(node->attrs(), "send_device", &send_device)); in GetRendezvousKey()
132 return Rendezvous::CreateKey(send_device, send_device_incarnation, in GetRendezvousKey()
|
| D | step_stats_collector.cc | 99 string send_device; in Done() local
100 TF_CHECK_OK(GetNodeAttr(attrs, "send_device", &send_device)); in Done()
102 tensor_name, " @", send_device, ")"); in Done()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | cluster_function_library_runtime_test.cc | 146 key: "send_device" in TEST_F()
187 key: "send_device" in TEST_F()
307 key: "send_device" in TEST_F()
349 key: "send_device" in TEST_F()
|
| D | master_session.cc | 400 string send_device; in TrackFeedsAndFetches() local
401 TF_CHECK_OK(GetNodeAttr(ndef, "send_device", &send_device)); in TrackFeedsAndFetches()
409 Rendezvous::CreateKey(send_device, send_device_incarnation, in TrackFeedsAndFetches()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_group_metadata.cc | 470 auto send_device = GetInstructionDevice(*channel.send); in VerifyChannelInstructions() local
472 if (!send_device) { in VerifyChannelInstructions()
480 if (*send_device != *send_done_device) { in VerifyChannelInstructions()
484 channel.id, *send_device, *send_done_device); in VerifyChannelInstructions()
498 if (*send_device == *recv_device) { in VerifyChannelInstructions()
501 channel.id, *send_device); in VerifyChannelInstructions()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | Recv.pbtxt | 16 name: "send_device"
|
| D | Send.pbtxt | 16 name: "send_device"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | fold-switch.mlir | 297 …nst/_0", recv_device = "/job:localhost/replica:0/task:0/device:CPU:0", send_device = "/job:localho…
301 …nst/_1", recv_device = "/job:localhost/replica:0/task:0/device:CPU:0", send_device = "/job:localho…
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 951 const string& send_device = GetNodeAttrString(*ndef, "send_device"); in SetIncarnation() local
952 if (send_device.empty()) { in SetIncarnation()
959 incarnation = opts.get_incarnation(send_device); in SetIncarnation()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute.cc | 1034 Device* send_device = h->DeviceOrHostCPU(*ctx); in EagerCopyToDevice() local
1036 bool sender_is_local = send_device->IsLocal(); in EagerCopyToDevice()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler_test.cc | 944 key: "send_device" in CreateGrapplerItemWithSendRecv()
979 key: "send_device" in CreateGrapplerItemWithSendRecv()
1037 key: "send_device" in CreateGrapplerItemWithRecvWithoutSend()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 3169 …argspec: "args=[\'tensor_type\', \'tensor_name\', \'send_device\', \'send_device_incarnation\', \'…
3757 …argspec: "args=[\'tensor\', \'tensor_name\', \'send_device\', \'send_device_incarnation\', \'recv_…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 3169 …argspec: "args=[\'tensor_type\', \'tensor_name\', \'send_device\', \'send_device_incarnation\', \'…
3757 …argspec: "args=[\'tensor\', \'tensor_name\', \'send_device\', \'send_device_incarnation\', \'recv_…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 21777 func Send(scope *Scope, tensor tf.Output, tensor_name string, send_device string, send_device_incar…
21781 …attrs := map[string]interface{}{"tensor_name": tensor_name, "send_device": send_device, "send_devi…
41308 func Recv(scope *Scope, tensor_type tf.DataType, tensor_name string, send_device string, send_devic…
41312 …"tensor_type": tensor_type, "tensor_name": tensor_name, "send_device": send_device, "send_device_i…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 33643 name: "send_device"
40728 name: "send_device"
|