Searched defs:dim2_ (Results 1 – 2 of 2) sorted by relevance
315 auto dim2_ = maybe_wrap_dim(dim2, logical_rank) + 1; in diagonal_batching_rule() local
1054 Tensor diagonal(const Tensor& self, int64_t offset, int64_t dim1_, int64_t dim2_) { in diagonal()1122 Tensor diag_embed(const Tensor& self, int64_t offset, int64_t dim1_, int64_t dim2_) { in diag_embed()