Searched refs:ThreadDim (Results 1 – 16 of 16) 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()
186 ThreadDim thread_dim_limit_;
253 void set_thread_dim_limit(const ThreadDim &value) { in set_thread_dim_limit()
338 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 | 71 const ThreadDim &thread_dim = thread_dim_limit(); in ToMap()
130 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 | 190 virtual port::Status Launch(Stream *stream, const ThreadDim &thread_dims, in Launch()
|
| D | stream_executor_pimpl.h | 474 port::Status Launch(Stream *stream, const ThreadDim &thread_dims,
|
| D | stream_executor_pimpl.cc | 465 const ThreadDim &thread_dims, in Launch()
|
| D | stream.h | 185 Stream &ThenLaunch(ThreadDim thread_dims, BlockDim block_dims,
2119 inline Stream &Stream::ThenLaunch(ThreadDim thread_dims, BlockDim block_dims, in ThenLaunch()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_executor.h | 72 port::Status Launch(Stream* stream, const ThreadDim& thread_dims,
80 const ThreadDim& thread_dims, GpuFunctionHandle func);
87 const ThreadDim& thread_dims, GpuFunctionHandle func);
267 void VlogOccupancyInfo(const KernelBase& kernel, const ThreadDim& thread_dims,
|
| D | redzone_allocator.cc | 228 stream->ThenLaunch(ThreadDim(threads_per_block), BlockDim(block_count), in RunRedzoneChecker()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 296 port::Status GpuExecutor::Launch(Stream* stream, const ThreadDim& thread_dims, in Launch()
351 const ThreadDim& thread_dims, in CalculateOccupancy()
361 const ThreadDim& thread_dims, in CompareOccupancy()
420 const ThreadDim& thread_dims, in VlogOccupancyInfo()
857 ThreadDim thread_dim_limit; in CreateDeviceDescription()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 400 port::Status GpuExecutor::Launch(Stream* stream, const ThreadDim& thread_dims, in Launch()
440 const ThreadDim& thread_dims, in VlogOccupancyInfo()
484 const ThreadDim& thread_dims, in CalculateOccupancy()
501 const ThreadDim& thread_dims, in CompareOccupancy()
1005 ThreadDim thread_dim_limit; in CreateDeviceDescription()
|
| /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/compiler/xla/service/interpreter/ |
| D | executor.h | 64 port::Status Launch(Stream *stream, const ThreadDim &thread_dims, in Launch()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | buffer_comparator.cc | 624 se::ThreadDim(thread_counts.x, thread_counts.y, thread_counts.z), in DeviceCompare()
|
| D | stream_executor_util.cc | 221 stream, se::ThreadDim(thread_counts.x, thread_counts.y, thread_counts.z), in ExecuteKernelOnStream()
|