Searched defs:thread_dims (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | trace_listener.h | 49 virtual void LaunchSubmit(Stream* stream, const ThreadDim& thread_dims, in LaunchSubmit()
|
| D | stream_executor_internal.h | 190 virtual port::Status Launch(Stream *stream, const ThreadDim &thread_dims, in Launch()
|
| D | stream.h | 2304 inline Stream &Stream::ThenLaunch(ThreadDim thread_dims, BlockDim block_dims, in ThenLaunch()
|
| D | stream_executor_pimpl.cc | 472 const ThreadDim &thread_dims, in Launch()
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.h | 57 port::Status Launch(Stream *stream, const ThreadDim &thread_dims, in Launch()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 396 port::Status GpuExecutor::Launch(Stream* stream, const ThreadDim& thread_dims, in Launch()
437 const ThreadDim& thread_dims, in VlogOccupancyInfo()
481 const ThreadDim& thread_dims, in CalculateOccupancy()
498 const ThreadDim& thread_dims, in CompareOccupancy()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 292 port::Status GpuExecutor::Launch(Stream* stream, const ThreadDim& thread_dims, in Launch()
347 const ThreadDim& thread_dims, in CalculateOccupancy()
357 const ThreadDim& thread_dims, in CompareOccupancy()
416 const ThreadDim& thread_dims, in VlogOccupancyInfo()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.h | 64 port::Status Launch(Stream *stream, const ThreadDim &thread_dims, in Launch()
|