| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_copy_node.cc | 78 Device* recv_device, uint64 recv_op_id) in RemoteCopyNode() argument
84 recv_device_(recv_device), in RemoteCopyNode()
244 Device* recv_device = recv_device_; in RunRemoteRecv() local
247 [captured_state, response, recv_device, context_view_id, in RunRemoteRecv()
251 response->queue_response(0).shape(0), recv_device, in RunRemoteRecv()
261 captured_state->dst()->PoisonRemote(s, recv_device, context_view_id); in RunRemoteRecv()
274 Device* recv_device = ctx_->CanonicalDevice(recv_device_); in StartRecv() local
276 captured_state_->dst()->Poison(status, recv_device); in StartRecv()
297 captured_state_->dst()->Poison(status, recv_device); in StartRecv()
302 captured_state_->dst()->SetTensor(std::move(outputs[0]), recv_device); in StartRecv()
[all …]
|
| D | remote_copy_node.h | 64 TensorHandle* dst, Device* recv_device, uint64 recv_op_id);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sendrecv_ops.cc | 30 const string& recv_device, in GetRendezvousKeyPrefix() argument
35 recv_device, ";", tensor_name); in GetRendezvousKeyPrefix()
61 string recv_device; in SendOp() local
62 OP_REQUIRES_OK(ctx, ctx->GetAttr("recv_device", &recv_device)); in SendOp()
69 key_prefix_ = GetRendezvousKeyPrefix(send_device, recv_device, in SendOp()
140 string recv_device; in RecvOp() local
141 OP_REQUIRES_OK(ctx, ctx->GetAttr("recv_device", &recv_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 | 29 name: "recv_device"
43 summary: "Receives the named tensor from send_device on recv_device."
|
| D | api_def_Send.pbtxt | 29 name: "recv_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 | 143 key: "recv_device" in TEST_F()
184 key: "recv_device" in TEST_F()
304 key: "recv_device" in TEST_F()
346 key: "recv_device" in TEST_F()
|
| D | master_session.cc | 404 string recv_device; in TrackFeedsAndFetches() local
405 TF_CHECK_OK(GetNodeAttr(ndef, "recv_device", &recv_device)); in TrackFeedsAndFetches()
412 recv_device, name, FrameAndIter(0, 0)); in TrackFeedsAndFetches()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | kernel_benchmark_testlib.cc | 150 string recv_device; in GetRendezvousKey() local
151 TF_CHECK_OK(GetNodeAttr(node->attrs(), "recv_device", &recv_device)); in GetRendezvousKey()
158 recv_device, tensor_name, FrameAndIter(0, 0)); in GetRendezvousKey()
|
| D | immutable_executor_state.cc | 114 string recv_device; in Initialize() local
116 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "recv_device", &recv_device)); in Initialize()
117 if (send_device != recv_device) { in Initialize()
|
| D | step_stats_collector.cc | 92 string recv_device; in Done() local
93 TF_CHECK_OK(GetNodeAttr(attrs, "recv_device", &recv_device)); in Done()
95 tensor_name, " @", recv_device, ")"); in Done()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | Recv.pbtxt | 24 name: "recv_device"
|
| D | Send.pbtxt | 24 name: "recv_device"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | Send.pbtxt | 24 name: "recv_device"
|
| D | Recv.pbtxt | 24 name: "recv_device"
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_group_metadata.cc | 486 auto recv_device = GetInstructionDevice(*channel.recv); in VerifyChannelInstructions() local
492 if (*recv_device != *recv_done_device) { in VerifyChannelInstructions()
496 channel.id, *recv_device, *recv_done_device); in VerifyChannelInstructions()
498 if (*send_device == *recv_device) { in VerifyChannelInstructions()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler_test.cc | 993 key: "recv_device" in CreateGrapplerItemWithSendRecv()
1044 key: "recv_device" in CreateGrapplerItemWithSendRecv()
1102 key: "recv_device" in CreateGrapplerItemWithRecvWithoutSend()
3031 const string recv_device = kCPU1; in TEST_F() local
3035 node->set_device(recv_device); in TEST_F()
3037 (*attr)["recv_device"].set_s(recv_device); in TEST_F()
3040 (*attr)["recv_device"].set_s(recv_device); in TEST_F()
|
| /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.";
10511 StrAttr:$recv_device,
13344 let summary = "Sends the named tensor from send_device to recv_device.";
13352 StrAttr:$recv_device,
18422 let summary = "Receives the named tensor from send_device on recv_device.";
18433 StrAttr:$recv_device,
18445 let summary = "Sends the named tensor from send_device to recv_device.";
18458 StrAttr:$recv_device,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 3397 …pe\', \'tensor_name\', \'send_device\', \'send_device_incarnation\', \'recv_device\', \'client_ter…
4033 …or\', \'tensor_name\', \'send_device\', \'send_device_incarnation\', \'recv_device\', \'client_ter…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 3397 …pe\', \'tensor_name\', \'send_device\', \'send_device_incarnation\', \'recv_device\', \'client_ter…
4033 …or\', \'tensor_name\', \'send_device\', \'send_device_incarnation\', \'recv_device\', \'client_ter…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 34602 …tensor_name string, send_device string, send_device_incarnation int64, recv_device string, optiona…
34606 …vice": send_device, "send_device_incarnation": send_device_incarnation, "recv_device": recv_device}
42387 …tensor_name string, send_device string, send_device_incarnation int64, recv_device string, optiona…
42391 …vice": send_device, "send_device_incarnation": send_device_incarnation, "recv_device": recv_device}
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 36379 name: "recv_device"
45511 name: "recv_device"
|