Home
last modified time | relevance | path

Searched refs:batch_iterator_with_broadcasting (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DLinearAlgebraUtils.h197 void batch_iterator_with_broadcasting(const Tensor& a, const Tensor& b, const func_t& f) { in batch_iterator_with_broadcasting() function
H A DBatchLinearAlgebraKernel.cpp579 at::native::batch_iterator_with_broadcasting<scalar_t>(A, B, in apply_lstsq()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/
H A DBatchLinearAlgebra.cpp2612 batch_iterator_with_broadcasting<scalar_t>(a, b, in apply_gels()