Searched defs:self_dim (Results 1 – 6 of 6) sorted by relevance
109 ssize_t self_dim = self.dim() ? self.dim() : 1; in repeat_internal() local195 ssize_t self_dim = self.dim() ? self.dim() : 1; in repeat_tensor() local
24 int64_t self_dim = self.sizes().size(); in calc_out_mean_sizes() local
239 int64_t self_dim = self_ptr->dim(), mat2_dim = mat2_ptr->dim(); in matmul_nested() local
447 auto self_dim = self.dim(); in expand_batch_rule() local
691 auto self_dim = self.dim(); in prelu() local
3828 size_t self_dim = self.dim(); in movedim() local