Searched defs:logical_dim (Results 1 – 10 of 10) sorted by relevance
109 auto logical_dim = self.dim(); in squeeze_dims__batching_rule() local191 auto logical_dim = self.dim(); in unsqueeze__batching_rule() local211 auto logical_dim = self.dim(); in transpose__batching_rule() local
44 int64_t getPhysicalDim(const Tensor& tensor, bool has_batch_dim, int64_t logical_dim) { in getPhysicalDim()
130 auto logical_dim = rankWithoutBatchDim(self, self_bdim); in boxed_reduction_batch_rule() local
52 auto logical_dim = rankWithoutBatchDim(maybe_tensor.value(), indices_bdims[i]); in get_max_index_logical_dim() local
60 for (int logical_dim = 0; logical_dim < shape.rank(); logical_dim++) { in HandleReduce() local
99 int64_t logical_dim = in HandleReduce() local
81 auto logical_dim = self.dim(); in maybe_movedim() local
107 auto logical_dim = self.dim(); in _movedim() local
554 int64_t logical_dim = Minor(shape.layout(), minor); in LinearIndex() local
1457 enum isl_surf_dim logical_dim, in isl_surf_choose_dim_layout()