Searched defs:pixel_unshuffle (Results 1 – 5 of 5) sorted by relevance
15 inline Tensor pixel_unshuffle(const Tensor& input, int64_t downscale_factor) { in pixel_unshuffle() function39 inline Tensor pixel_unshuffle( in pixel_unshuffle() function
433 at::Tensor LazyNativeFunctions::pixel_unshuffle( in pixel_unshuffle() function in torch::lazy::LazyNativeFunctions
1248 def pixel_unshuffle(self: Tensor, downscale_factor: int): function
4065 pixel_unshuffle = _add_docstr( variable
4198 def pixel_unshuffle(g: jit_utils.GraphContext, self, downscale_factor): function