Searched refs:ThreadDim (Results 1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | launch_dim.h | 54 struct ThreadDim : public Dim3D { struct
55 explicit ThreadDim(uint64_t x = 1, uint64 y = 1, uint64 z = 1)
|
| D | device_description.h | 211 const ThreadDim &thread_dim_limit() const { return thread_dim_limit_; } in thread_dim_limit()
316 ThreadDim thread_dim_limit_;
379 void set_thread_dim_limit(const ThreadDim &value) { in set_thread_dim_limit()
461 const ThreadDim &thread_dim);
|
| D | device_description.cc | 67 const ThreadDim &thread_dim = thread_dim_limit(); in ToMap()
118 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_internal.h | 187 virtual port::Status Launch(Stream* stream, const ThreadDim& thread_dims, in Launch()
|
| D | stream_executor_pimpl.h | 478 port::Status Launch(Stream* stream, const ThreadDim& thread_dims,
|
| D | stream.h | 191 port::Status ThenLaunch(ThreadDim thread_dims, BlockDim block_dims,
1623 inline port::Status Stream::ThenLaunch(ThreadDim thread_dims, in ThenLaunch()
|
| D | stream_executor_pimpl.cc | 475 const ThreadDim& thread_dims, in Launch()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/ |
| D | gpu_executor.h | 123 port::Status Launch(Stream* stream, const ThreadDim& thread_dims,
131 const ThreadDim& thread_dims, GpuFunctionHandle func);
138 const ThreadDim& thread_dims, GpuFunctionHandle func);
342 void VlogOccupancyInfo(const KernelBase& kernel, const ThreadDim& thread_dims,
|
| D | redzone_allocator.cc | 230 ThreadDim(threads_per_block), BlockDim(block_count), comparison_kernel, in RunRedzoneChecker()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 298 port::Status GpuExecutor::Launch(Stream* stream, const ThreadDim& thread_dims, in Launch()
353 const ThreadDim& thread_dims, in CalculateOccupancy()
363 const ThreadDim& thread_dims, in CompareOccupancy()
422 const ThreadDim& thread_dims, in VlogOccupancyInfo()
910 ThreadDim thread_dim_limit; in CreateDeviceDescription()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 441 port::Status GpuExecutor::Launch(Stream* stream, const ThreadDim& thread_dims, in Launch()
482 const ThreadDim& thread_dims, in VlogOccupancyInfo()
526 const ThreadDim& thread_dims, in CalculateOccupancy()
543 const ThreadDim& thread_dims, in CompareOccupancy()
1044 ThreadDim thread_dim_limit; in CreateDeviceDescription()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | precompiled_kernels.cc | 176 stream->ThenLaunch(se::ThreadDim(kThreads, 1, 1), in MakeBatchPointers()
|
| D | buffer_comparator.cc | 680 se::ThreadDim(thread_counts.x, thread_counts.y, thread_counts.z), in DeviceCompare()
|
| D | stream_executor_util.cc | 377 stream, se::ThreadDim(thread_counts.x, thread_counts.y, thread_counts.z), in ExecuteKernelOnStream()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/host/ |
| D | host_gpu_executor.h | 57 port::Status Launch(Stream* stream, const ThreadDim& thread_dims, in Launch()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.h | 63 port::Status Launch(Stream *stream, const ThreadDim &thread_dims, in Launch()
|