Searched defs:pixel_shuffle (Results 1 – 6 of 6) sorted by relevance
11 inline Tensor pixel_shuffle(const Tensor& input, int64_t upscale_factor) { in pixel_shuffle() function33 inline Tensor pixel_shuffle( in pixel_shuffle() function
427 at::Tensor LazyNativeFunctions::pixel_shuffle( in pixel_shuffle() function in torch::lazy::LazyNativeFunctions
1220 def pixel_shuffle(self: Tensor, upscale_factor: int): function
327 def pixel_shuffle(g: jit_utils.GraphContext, self, upscale_factor): function
4126 def pixel_shuffle(g: jit_utils.GraphContext, self, upscale_factor): function
4042 pixel_shuffle = _add_docstr( variable