Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DBatchLinearAlgebra.cpp1582 static bool _may_require_fw_or_bw_grad(const Tensor& input) { in _may_require_fw_or_bw_grad() function
2778 /*compute_v=*/_may_require_fw_or_bw_grad(A))); in linalg_eigvalsh()
3108 if (_may_require_fw_or_bw_grad(input)) { in linalg_eigvals()
3220 /*compute_uv=*/_may_require_fw_or_bw_grad(A), in linalg_svdvals()