Searched refs:tensor_logical_rank (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesHelper.cpp | 75 auto tensor_logical_rank = rankWithoutBatchDim(tensor, has_bdim); in maybePadToLogicalRank() local
76 if (tensor_logical_rank >= logical_rank) { in maybePadToLogicalRank()
80 for (int64_t i = 0; i < logical_rank - tensor_logical_rank; i++) { in maybePadToLogicalRank()
176 auto tensor_logical_rank = rankWithoutBatchDim(tensor, tensor_batch_dim); in _binary_pointwise_helper() local
178 auto max_logical_rank = std::max(tensor_logical_rank, other_logical_rank); in _binary_pointwise_helper()
185 auto tensor_is_logical_scalar = (tensor_logical_rank == 0 && tensor_batch_dim.has_value()); in _binary_pointwise_helper()
|
| H A D | BatchRulesBinaryOps.cpp | 105 auto tensor_logical_rank = rankWithoutBatchDim(tensor, tensor_batch_dim); in binary_pointwise_inplace_batch_rule() local
107 auto max_logical_rank = std::max(tensor_logical_rank, other_logical_rank); in binary_pointwise_inplace_batch_rule()
127 auto tensor_logical_rank = rankWithoutBatchDim(tensor, tensor_batch_dim); in comparison_pointwise_batch_rule() local
129 auto max_logical_rank = std::max(tensor_logical_rank, other_logical_rank); in comparison_pointwise_batch_rule()
149 auto tensor_logical_rank = rankWithoutBatchDim(self, self_bdim); in where_self_batch_rule() local
151 …auto max_logical_rank = std::max({tensor_logical_rank, other_logical_rank, condition_logical_rank}… in where_self_batch_rule()
|
| H A D | BatchRulesNorm.cpp | 33 auto tensor_logical_rank = rankWithoutBatchDim(tensor, has_bdim); in padRight() local
34 if (tensor_logical_rank >= logical_rank) { in padRight()
38 for (int64_t i = 0; i < logical_rank - tensor_logical_rank; i++) { in padRight()
|
| H A D | BatchRulesScatterOps.cpp | 1024 auto tensor_logical_rank = rankWithoutBatchDim(self, self_bdim); in binary_pointwise_align() local
1026 auto max_logical_rank = std::max(tensor_logical_rank, other_logical_rank); in binary_pointwise_align()
|