| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_internal.cc | 48 bool StreamExecutorInterface::HostCallback(Stream* stream, in HostCallback() function in stream_executor::internal::StreamExecutorInterface
50 return HostCallback( in HostCallback()
|
| D | stream_executor_internal.h | 242 virtual bool HostCallback(Stream *stream, std::function<void()> callback);
243 virtual bool HostCallback(Stream *stream,
|
| D | stream_executor_pimpl.cc | 744 bool StreamExecutor::HostCallback(Stream *stream, in HostCallback() function in stream_executor::StreamExecutor
746 return implementation_->HostCallback(stream, std::move(callback)); in HostCallback()
749 bool StreamExecutor::HostCallback(Stream *stream, in HostCallback() function in stream_executor::StreamExecutor
751 return implementation_->HostCallback(stream, std::move(callback)); in HostCallback()
|
| D | stream_executor_pimpl.h | 563 bool HostCallback(Stream *stream, std::function<void()> callback);
568 bool HostCallback(Stream *stream, std::function<port::Status()> callback);
|
| D | stream.cc | 5123 CheckError(parent_->HostCallback(this, std::move(callback))); in ThenDoHostCallback()
5135 CheckError(parent_->HostCallback(this, std::move(callback))); in ThenDoHostCallbackWithStatus()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.cc | 81 bool XlaInterpreterExecutor::HostCallback( in HostCallback() function in stream_executor::interpreter::XlaInterpreterExecutor
|
| D | executor.h | 128 bool HostCallback(Stream *stream,
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.h | 106 bool HostCallback(Stream *stream,
|
| D | host_gpu_executor.cc | 150 bool HostExecutor::HostCallback(Stream *stream, in HostCallback() function in stream_executor::host::HostExecutor
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_executor.h | 155 bool HostCallback(Stream* stream,
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 559 bool GpuExecutor::HostCallback(Stream* stream, in HostCallback() function in stream_executor::gpu::GpuExecutor
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 678 bool GpuExecutor::HostCallback(Stream* stream, in HostCallback() function in stream_executor::gpu::GpuExecutor
|