| /external/tensorflow/tensorflow/stream_executor/ |
| stream_executor_internal.cc | 46 // The default implementation just calls the other HostCallback method.
48 bool StreamExecutorInterface::HostCallback(Stream* stream,
50 return HostCallback(
|
| stream_executor_internal.h | 242 virtual bool HostCallback(Stream *stream, std::function<void()> callback);
243 virtual bool HostCallback(Stream *stream,
|
| stream_executor_pimpl.cc | 744 bool StreamExecutor::HostCallback(Stream *stream,
746 return implementation_->HostCallback(stream, std::move(callback));
749 bool StreamExecutor::HostCallback(Stream *stream,
751 return implementation_->HostCallback(stream, std::move(callback));
|
| stream_executor_pimpl.h | 563 bool HostCallback(Stream *stream, std::function<void()> callback);
568 bool HostCallback(Stream *stream, std::function<port::Status()> callback);
[all...] |
| stream.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| executor.cc | 81 bool XlaInterpreterExecutor::HostCallback(
|
| executor.h | 128 bool HostCallback(Stream *stream,
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| host_gpu_executor.cc | 150 bool HostExecutor::HostCallback(Stream *stream,
|
| host_gpu_executor.h | 106 bool HostCallback(Stream *stream,
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| gpu_executor.h | 155 bool HostCallback(Stream* stream,
257 // data: User-provided callback provided to HostCallback() above, captured
259 // HostCallback() and owned and deleted by this call.
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_gpu_executor.cc | 678 bool GpuExecutor::HostCallback(Stream* stream,
[all...] |
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| rocm_gpu_executor.cc | 559 bool GpuExecutor::HostCallback(Stream* stream,
[all...] |
