Home
last modified time | relevance | path

Searched defs:logical_rank (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesViews.cpp296 auto logical_rank = rankWithoutBatchDim(self, bdim); in roll_batch_rule() local
312 auto logical_rank = rankWithoutBatchDim(self, self_bdim); in diagonal_batching_rule() local
323 auto logical_rank = rankWithoutBatchDim(grad_input, grad_input_bdim); in diagonal_backward_batch_rule() local
395 auto logical_rank = rankWithoutBatchDim(grad_input, grad_input_bdim); in select_backward_batch_rule() local
408 auto logical_rank = rankWithoutBatchDim(grad_input, grad_input_bdim); in slice_backward_batch_rule() local
483 auto logical_rank = rankWithoutBatchDim(self, self_bdim); in unfold_batch_rule() local
500 auto logical_rank = rankWithoutBatchDim(self, self_bdim); in narrow_copy_batch_rule() local
514 auto logical_rank = rankWithoutBatchDim(self, self_bdim); in unsafe_split_batch_rule() local
521 auto logical_rank = rankWithoutBatchDim(self, self_bdim); in diag_embed_batch_rule() local
H A DBatchRulesPooling.cpp21 auto logical_rank = rankWithoutBatchDim(self, self_bdim); in max_pool_with_indices_batch_rule_helper() local
H A DBatchRulesHelper.cpp71 …aybePadToLogicalRank(const Tensor& tensor, std::optional<int64_t> has_bdim, int64_t logical_rank) { in maybePadToLogicalRank()
H A DBatchRulesNorm.cpp31 static Tensor padRight(const Tensor& tensor, std::optional<int64_t> has_bdim, int64_t logical_rank)… in padRight()