Searched defs:max_logical_rank (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesBinaryOps.cpp | 107 auto max_logical_rank = std::max(tensor_logical_rank, other_logical_rank); in binary_pointwise_inplace_batch_rule() local
129 auto max_logical_rank = std::max(tensor_logical_rank, other_logical_rank); in comparison_pointwise_batch_rule() local
151 …auto max_logical_rank = std::max({tensor_logical_rank, other_logical_rank, condition_logical_rank}… in where_self_batch_rule() local
191 const auto max_logical_rank = std::max(self_logical_rank, mask.dim()); in masked_select_batch_rule() local
212 const auto max_logical_rank = std::max(self_logical_rank, mask.dim()); in masked_select_backward_batch_rule() local
|
| H A D | BatchRulesHelper.cpp | 178 auto max_logical_rank = std::max(tensor_logical_rank, other_logical_rank); in _binary_pointwise_helper() local
|
| H A D | BatchRulesRandomness.cpp | 80 auto max_logical_rank = std::max(self_logical_rank, other_logical_rank); in bernoulli_inplace_Tensor_batching_rule() local
|
| H A D | BatchRulesLinearAlgebra.cpp | 368 const auto max_logical_rank = std::max(A_logical_rank, B_logical_rank); in solve_ex_batch_rule() local
|
| H A D | BatchRulesScatterOps.cpp | 1026 auto max_logical_rank = std::max(tensor_logical_rank, other_logical_rank); in binary_pointwise_align() local
|