Home
last modified time | relevance | path

Searched defs:total_work_block (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DFunctionOfAMatrixUtilsKernel.cu17 constexpr int total_work_block = n_threads * n_elems_per_thread; in C10_LAUNCH_BOUNDS_2() local
36 constexpr int total_work_block = n_threads * n_elems_per_thread; in _lauch_kernel() local
H A DLinearAlgebra.cu78 constexpr int total_work_block = n_threads * n_elems_per_thread; in C10_LAUNCH_BOUNDS_2() local
97 constexpr int total_work_block = n_threads * n_elems_per_thread; in _launch_kernel() local
H A DUnfoldBackwardKernel.cu25 constexpr int total_work_block = n_threads * n_elems_per_thread; in C10_LAUNCH_BOUNDS_2() local
44 constexpr int total_work_block = n_threads * n_elems_per_thread; in _launch_unfold_backward_kernel() local