Searched defs:self_logical_rank (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesScatterOps.cpp | 680 auto self_logical_rank = rankWithoutBatchDim(self, self_bdim); in scatter_batch_rule() local
712 auto self_logical_rank = rankWithoutBatchDim(self, self_bdim); in scatter_batch_rule() local
797 auto self_logical_rank = rankWithoutBatchDim(self, self_bdim); in gather_batch_rule() local
918 auto self_logical_rank = rankWithoutBatchDim(self, self_bdim); in diagonal_scatter_batch_rule() local
939 const auto self_logical_rank = rankWithoutBatchDim(self, self_bdim); in index_add_batch_rule_impl() local
1054 int64_t self_logical_rank, in index_fill_batch_rule_helper()
1094 const auto self_logical_rank = rankWithoutBatchDim(self, self_bdim); in index_fill_int_scalar_batch_rule_impl() local
1145 const auto self_logical_rank = rankWithoutBatchDim(self, self_bdim); in index_fill_int_tensor_batch_rule_impl() local
|
| H A D | BatchRulesLinearAlgebra.cpp | 72 auto self_logical_rank = rankWithoutBatchDim(self, self_bdim); in mv_batch_rule() local
85 auto self_logical_rank = rankWithoutBatchDim(self, self_bdim); in mm_batch_rule() local
100 auto self_logical_rank = rankWithoutBatchDim(self, self_bdim); in bmm_batch_rule() local
|
| H A D | BatchRulesBinaryOps.cpp | 190 const auto self_logical_rank = rankWithoutBatchDim(self, self_bdim); in masked_select_batch_rule() local
211 const auto self_logical_rank = rankWithoutBatchDim(self, self_bdim); in masked_select_backward_batch_rule() local
|
| H A D | BatchRulesReduceOps.cpp | 328 auto self_logical_rank = rankWithoutBatchDim(self, self_bdim); in searchsorted_batch_rule() local
|
| H A D | BatchRulesRandomness.cpp | 78 auto self_logical_rank = rankWithoutBatchDim(self_value, self_bdim); in bernoulli_inplace_Tensor_batching_rule() local
|