Searched refs:check_pixel_shuffle_shapes (Results 1 – 3 of 3) sorted by relevance
24 check_pixel_shuffle_shapes(self, upscale_factor); in pixel_shuffle_cpu()76 check_pixel_shuffle_shapes(self, upscale_factor); in math_pixel_shuffle()
7 inline void check_pixel_shuffle_shapes(const Tensor& self, int64_t upscale_factor) { in check_pixel_shuffle_shapes() function
19 check_pixel_shuffle_shapes(self, factor);