Home
last modified time | relevance | path

Searched defs:max_pivot (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DLinearAlgebra.cu106 … unpack_pivots_cuda_kernel(TensorIterator& iter, const int64_t dim_size, const int64_t max_pivot) { in unpack_pivots_cuda_kernel()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DBatchLinearAlgebraKernel.cpp1104 void unpack_pivots_cpu_kernel(TensorIterator& iter, const int64_t dim_size, const int64_t max_pivot… in unpack_pivots_cpu_kernel()