Searched defs:tensor_other (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesLinearAlgebra.cpp | 239 … const auto tensor_other = check_inputs_and_reshape_inputs(first, first_bdim, second, second_bdim); in apply_one() local
249 … const auto tensor_other = check_inputs_and_reshape_inputs(first, first_bdim, second, second_bdim); in apply_two() local
341 …const auto tensor_other = _binary_pointwise_helper(self, self_bdim, A, A_bdim, /*do_type_promotion… in cholesky_solve_batch_rule() local
435 …const auto tensor_other = _binary_pointwise_helper(self, self_bdim, b, b_bdim, /*do_type_promotion… in linalg_lstsq_batch_rule() local
|
| H A D | BatchRulesBinaryOps.cpp | 22 auto tensor_other = _binary_pointwise_helper( in _binary_pointwise_batch_rule() local
168 const auto tensor_other = _binary_pointwise_helper(grad_out, grad_out_bdim, input, input_bdim); in gelu_backward_batch_rule() local
|