Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/
H A DBatchLinearAlgebraLib.cpp257 auto S_data = S.data_ptr<value_t>(); in apply_svd_cusolver_gesvd() local
373 auto S_data = S.data_ptr<value_t>(); in apply_svd_cusolver_gesvdj() local
459 auto S_data = S.data_ptr<value_t>(); in apply_svd_cusolver_gesvdjBatched() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A Dint8mm_kernel.cpp368 const auto* S_data = scales.const_data_ptr<T>(); in int8pack_mm_kernel_() local
H A Dint4mm_kernel.cpp709 const auto* S_data = qScaleAndZeros.const_data_ptr<T>(); in int4pack_mm_kernel_() local