| /external/tensorflow/tensorflow/core/kernels/ |
| D | sendrecv_ops.cc | 29 static string GetRendezvousKeyPrefix(const string& send_device, in GetRendezvousKeyPrefix() argument
33 return strings::StrCat(send_device, ";", in GetRendezvousKeyPrefix()
59 string send_device; in SendOp() local
60 OP_REQUIRES_OK(ctx, ctx->GetAttr("send_device", &send_device)); in SendOp()
69 key_prefix_ = GetRendezvousKeyPrefix(send_device, recv_device, in SendOp()
138 string send_device; in RecvOp() local
139 OP_REQUIRES_OK(ctx, ctx->GetAttr("send_device", &send_device)); in RecvOp()
148 key_prefix_ = GetRendezvousKeyPrefix(send_device, recv_device, in RecvOp()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| 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."
|
| 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."
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | cluster_function_library_runtime_test.cc | 149 key: "send_device" in TEST_F()
190 key: "send_device" in TEST_F()
310 key: "send_device" in TEST_F()
352 key: "send_device" in TEST_F()
|
| D | master_session.cc | 402 string send_device; in TrackFeedsAndFetches() local
403 TF_CHECK_OK(GetNodeAttr(ndef, "send_device", &send_device)); in TrackFeedsAndFetches()
411 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/common_runtime/ |
| D | kernel_benchmark_testlib.cc | 148 string send_device; in GetRendezvousKey() local
149 TF_CHECK_OK(GetNodeAttr(node->attrs(), "send_device", &send_device)); in GetRendezvousKey()
157 return Rendezvous::CreateKey(send_device, send_device_incarnation, in GetRendezvousKey()
|
| D | immutable_executor_state.cc | 113 string send_device; in Initialize() local
115 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "send_device", &send_device)); in Initialize()
117 if (send_device != recv_device) { in Initialize()
|
| 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/ops/compat/ops_history_v1/ |
| D | Recv.pbtxt | 16 name: "send_device"
|
| D | Send.pbtxt | 16 name: "send_device"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | Send.pbtxt | 16 name: "send_device"
|
| D | Recv.pbtxt | 16 name: "send_device"
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 953 const string& send_device = GetNodeAttrString(*ndef, "send_device"); in SetIncarnation() local
954 if (send_device.empty()) { in SetIncarnation()
961 incarnation = opts.get_incarnation(send_device); in SetIncarnation()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler.cc | 687 auto send_device = in CreateSendRecv() local
689 send->set_device(send_device); in CreateSendRecv()
|
| D | virtual_scheduler_test.cc | 999 key: "send_device" in CreateGrapplerItemWithSendRecv()
1050 key: "send_device" in CreateGrapplerItemWithSendRecv()
1108 key: "send_device" in CreateGrapplerItemWithRecvWithoutSend()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute.cc | 1219 auto send_device = h->DeviceOrHostCPU(*ctx); in EagerCopyToDevice() local
1220 bool sender_is_local = send_device->IsLocal(); in EagerCopyToDevice()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 10505 let summary = "Receives the named tensor from send_device on recv_device.";
10509 StrAttr:$send_device,
13344 let summary = "Sends the named tensor from send_device to recv_device.";
13350 StrAttr:$send_device,
18422 let summary = "Receives the named tensor from send_device on recv_device.";
18431 StrAttr:$send_device,
18445 let summary = "Sends the named tensor from send_device to recv_device.";
18456 StrAttr:$send_device,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 3397 …argspec: "args=[\'tensor_type\', \'tensor_name\', \'send_device\', \'send_device_incarnation\', \'…
4033 …argspec: "args=[\'tensor\', \'tensor_name\', \'send_device\', \'send_device_incarnation\', \'recv_…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 3397 …argspec: "args=[\'tensor_type\', \'tensor_name\', \'send_device\', \'send_device_incarnation\', \'…
4033 …argspec: "args=[\'tensor\', \'tensor_name\', \'send_device\', \'send_device_incarnation\', \'recv_…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 34602 func Send(scope *Scope, tensor tf.Output, tensor_name string, send_device string, send_device_incar…
34606 …attrs := map[string]interface{}{"tensor_name": tensor_name, "send_device": send_device, "send_devi…
42387 func Recv(scope *Scope, tensor_type tf.DataType, tensor_name string, send_device string, send_devic…
42391 …"tensor_type": tensor_type, "tensor_name": tensor_name, "send_device": send_device, "send_device_i…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 36371 name: "send_device"
45503 name: "send_device"
|