Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cuda/
DTensorModeKernel.cpp64 auto values_transposed = values.transpose(dim, ndim - 1); in mode_kernel_impl() local
81 values_transposed, indices_transposed, contiguous, slice_size, slices); in mode_kernel_impl()
92 values_transposed, indices_transposed, contiguous, dim, ndim); in mode_kernel_impl()