Searched refs:wait_until_ready (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_mgr.cc | 84 const bool wait_until_ready, in GetRemoteTensorHandle() argument
86 TF_RETURN_IF_ERROR(handle->RemoteAddress(handle->device(), wait_until_ready, in GetRemoteTensorHandle()
124 TensorHandle* in, const bool wait_until_ready, RemoteTensorHandle* out, in SerializeRemoteTensorHandle() argument
129 if (!in->RemoteAddress(device, wait_until_ready, &op_id, &output_num).ok()) { in SerializeRemoteTensorHandle()
132 GetRemoteTensorHandle(in, wait_until_ready, &op_id, &output_num)); in SerializeRemoteTensorHandle()
|
| D | remote_mgr.h | 70 TensorHandle* in, const bool wait_until_ready, RemoteTensorHandle* out,
91 const bool wait_until_ready, int64* op_id,
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorContractionThreadPool.h | 866 wait_until_ready(kernel_notifications[notification_id]);
902 wait_until_ready(kernel_notifications[future_id]);
940 wait_until_ready(kernel_notifications[i]);
994 wait_until_ready((*arg.lhs_notifications)[blockAId]);
|
| D | TensorDeviceThreadPool.h | 99 static EIGEN_STRONG_INLINE void wait_until_ready(SyncType* n) { in wait_until_ready() function
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle.h | 190 Status RemoteAddress(const Device* d, const bool wait_until_ready,
|
| D | tensor_handle.cc | 709 Status TensorHandle::RemoteAddress(const Device* d, const bool wait_until_ready, in RemoteAddress() argument
718 return mirror->second.OpIdAndOutputNum(wait_until_ready, op_id, in RemoteAddress()
731 return data.OpIdAndOutputNum(wait_until_ready, op_id, output_num); in RemoteAddress()
|
| D | execute.cc | 901 const bool wait_until_ready = op->is_function(); in EagerRemoteExecute() local
903 input, wait_until_ready, input_handle, input_device, in EagerRemoteExecute()
|