Searched refs:PoisonRemote (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_copy_node.cc | 225 captured_state_->dst()->PoisonRemote(status, recv_device_, context_view_id); in RunRemoteRecv()
237 captured_state_->dst()->PoisonRemote(status, recv_device_, context_view_id); in RunRemoteRecv()
261 captured_state->dst()->PoisonRemote(s, recv_device, context_view_id); in RunRemoteRecv()
360 captured_state_->dst()->PoisonRemote(s, recv_device_, context_view_id); in StartSendPackedHandle()
371 captured_state_->dst()->PoisonRemote(s, recv_device_, context_view_id); in StartSendPackedHandle()
379 captured_state_->dst()->PoisonRemote(s, recv_device_, context_view_id); in StartSendPackedHandle()
388 captured_state_->dst()->PoisonRemote(s, recv_device_, context_view_id); in StartSendPackedHandle()
409 captured_state->dst()->PoisonRemote(s, recv_device, context_view_id); in StartSendPackedHandle()
442 captured_state_->dst()->PoisonRemote(s, recv_device_, context_view_id); in StartRemoteSendTensor()
463 captured_state->dst()->PoisonRemote(s, recv_device, context_view_id); in StartRemoteSendTensor()
[all …]
|
| D | remote_execute_node.cc | 62 retvals[i]->PoisonRemote(s, device, context_view_id_); in RunAsync()
113 retvals[i]->PoisonRemote(status, device, context_view_id); in RunAsync()
|
| D | remote_execute_node.h | 104 handle->PoisonRemote(status, device_, context_view_id_); in Abort()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle.h | 212 void PoisonRemote(Status status, const Device* d, uint64 context_view_id);
|
| D | tensor_handle.cc | 886 void TensorHandle::PoisonRemote(Status status, const Device* d, in PoisonRemote() function in tensorflow::TensorHandle
|