Searched defs:wrap_dim (Results 1 – 6 of 6) sorted by relevance
28 auto wrap_dim = maybe_wrap_dim(dim, self.dim()); in TORCH_META_FUNC() local57 auto wrap_dim = maybe_wrap_dim(dim, input.dim()); in glu_backward_cpu_out() local
80 auto wrap_dim = maybe_wrap_dim(dim, self.dim()); in cumsum_cpu_kernel() local99 auto wrap_dim = maybe_wrap_dim(dim, self.dim()); in cumprod_cpu_kernel() local118 auto wrap_dim = maybe_wrap_dim(dim, self.dim()); in logcumsumexp_cpu_kernel() local
173 auto wrap_dim = maybe_wrap_dim(dim, self.dim()); in aminmax_kernel() local
32 auto wrap_dim = maybe_wrap_dim(dim, input.dim()); in glu_backward_cuda_out() local
63 const auto wrap_dim = maybe_wrap_dim(dim, self.dim()); in _logcumsumexp_out_cuda() local
2328 def wrap_dim(dim, ndims): function