Searched defs:sub_channels (Results 1 – 3 of 3) sorted by relevance
27 int64_t sub_channels = channels / (upscale_factor * upscale_factor); in cpu_pixel_shuffle() local69 int64_t sub_channels = channels / (upscale_factor * upscale_factor); in cpu_pixel_shuffle_channels_last() local122 int64_t sub_channels = input.size(-3); in cpu_pixel_unshuffle() local165 int64_t sub_channels = input.size(1); in cpu_pixel_unshuffle_channels_last() local
32 const auto sub_channels = channels / (S * S); in pixel_unshuffle_impl() local
28 const auto sub_channels = channels / (upscale_factor * upscale_factor); in pixel_shuffle_impl() local