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 | 191 virtual bool Launch(Stream *stream, const ThreadDim &thread_dims, in Launch()
|
| D | stream_executor_pimpl.h | 867 inline Stream &Stream::ThenLaunch(ThreadDim thread_dims, BlockDim block_dims, in ThenLaunch()
|
| D | stream_executor_pimpl.cc | 479 bool StreamExecutor::Launch(Stream *stream, const ThreadDim &thread_dims, in Launch()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 286 bool GpuExecutor::Launch(Stream* stream, const ThreadDim& thread_dims, in Launch()
347 const ThreadDim& thread_dims, in CalculateOccupancy()
357 const ThreadDim& thread_dims, in CompareOccupancy()
420 const ThreadDim& thread_dims, in VlogOccupancyInfo()
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.h | 57 bool Launch(Stream *stream, const ThreadDim &thread_dims, in Launch()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 420 bool GpuExecutor::Launch(Stream* stream, const ThreadDim& thread_dims, in Launch()
467 const ThreadDim& thread_dims, in VlogOccupancyInfo()
511 const ThreadDim& thread_dims, in CalculateOccupancy()
528 const ThreadDim& thread_dims, in CompareOccupancy()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.h | 65 bool Launch(Stream *stream, const ThreadDim &thread_dims, in Launch()
|