Searched refs:ThreadDim (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | device_description.h | 74 const ThreadDim &thread_dim_limit() const { return thread_dim_limit_; } in thread_dim_limit()
183 ThreadDim thread_dim_limit_;
247 void set_thread_dim_limit(const ThreadDim &value) { in set_thread_dim_limit()
332 const ThreadDim &thread_dim);
|
| D | launch_dim.h | 54 struct ThreadDim : public Dim3D { struct
55 explicit ThreadDim(uint64 x = 1, uint64 y = 1, uint64 z = 1)
|
| D | device_description.cc | 70 const ThreadDim &thread_dim = thread_dim_limit(); in ToMap()
127 const ThreadDim &thread_dim) { in ThreadDimOk()
|
| D | trace_listener.h | 49 virtual void LaunchSubmit(Stream* stream, const ThreadDim& thread_dims, in LaunchSubmit()
|
| D | stream_executor_pimpl.h | 452 bool Launch(Stream *stream, const ThreadDim &thread_dims,
867 inline Stream &Stream::ThenLaunch(ThreadDim thread_dims, BlockDim block_dims, in ThenLaunch()
|
| D | stream_executor_internal.h | 191 virtual bool Launch(Stream *stream, const ThreadDim &thread_dims, in Launch()
|
| D | stream_executor_pimpl.cc | 479 bool StreamExecutor::Launch(Stream *stream, const ThreadDim &thread_dims, in Launch()
|
| D | stream.h | 171 Stream &ThenLaunch(ThreadDim thread_dims, BlockDim block_dims,
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_executor.h | 72 bool Launch(Stream* stream, const ThreadDim& thread_dims,
80 const ThreadDim& thread_dims, GpuFunctionHandle func);
87 const ThreadDim& thread_dims, GpuFunctionHandle func);
269 void VlogOccupancyInfo(const KernelBase& kernel, const ThreadDim& thread_dims,
|
| /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()
904 ThreadDim thread_dim_limit; in PopulateDeviceDescription()
|
| /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()
1059 ThreadDim thread_dim_limit; in PopulateDeviceDescription()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | kernel_thunk.cc | 103 stream, se::ThreadDim(launch_dimensions.threads_per_block()), in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.h | 65 bool Launch(Stream *stream, const ThreadDim &thread_dims, in Launch()
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.h | 57 bool Launch(Stream *stream, const ThreadDim &thread_dims, in Launch()
|