Searched refs:_binary_pointwise_helper (Results 1 – 4 of 4) sorted by relevance
/external/pytorch/aten/src/ATen/functorch/ |
D | BatchRulesBinaryOps.cpp | 22 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()
|
D | BatchRulesLinearAlgebra.cpp | 232 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()
|
D | BatchRulesHelper.cpp | 171 std::tuple<Tensor, Tensor> _binary_pointwise_helper( in _binary_pointwise_helper() function
|
D | BatchRulesHelper.h | 471 std::tuple<Tensor, Tensor> _binary_pointwise_helper(
|