Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesLinearAlgebra.cpp464 const auto input_logical_num_bdims = input_logical_rank - 2; in atol_rtol_tensor_batch_rule() local
467 …const auto max_logical_bdims = std::max({input_logical_num_bdims, atol_logical_num_bdims, rtol_log… in atol_rtol_tensor_batch_rule()