Home
last modified time | relevance | path

Searched refs:dim_transpose (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/torch/distributed/tensor/_ops/
D_view_ops.py396 def dim_transpose(ndim: int, dim1: int, dim2: int) -> DimMap: function
475 torch.transpose: lambda input, dim0, dim1: dim_transpose(input.ndim, dim0, dim1),