Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/functorch/
DBatchRulesBinaryOps.cpp22 auto tensor_other = _binary_pointwise_helper( in _binary_pointwise_batch_rule()
168 const auto tensor_other = _binary_pointwise_helper(grad_out, grad_out_bdim, input, input_bdim); in gelu_backward_batch_rule()
245 auto x12 = _binary_pointwise_helper(x1_, x1_bdim, x2, x2_bdim); in cdist_backward_batch_rule()
283 auto self_and_other = _binary_pointwise_helper( in fill__Tensor_batch_rule()
DBatchRulesLinearAlgebra.cpp232 return _binary_pointwise_helper(first, first_bdim, second, second_bdim, false); in check_inputs_and_reshape_inputs()
341 …const auto tensor_other = _binary_pointwise_helper(self, self_bdim, A, A_bdim, /*do_type_promotion… in cholesky_solve_batch_rule()
411 …const auto self_other_bundled = _binary_pointwise_helper(self, self_bdim, other, other_bdim, false… in cross_batch_rule()
435 …const auto tensor_other = _binary_pointwise_helper(self, self_bdim, b, b_bdim, /*do_type_promotion… in linalg_lstsq_batch_rule()
DBatchRulesHelper.cpp171 std::tuple<Tensor, Tensor> _binary_pointwise_helper( in _binary_pointwise_helper() function
DBatchRulesHelper.h471 std::tuple<Tensor, Tensor> _binary_pointwise_helper(