Searched defs:downscale_factor (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | PixelShuffleKernel.cpp | 117 int64_t downscale_factor) { in cpu_pixel_unshuffle()
158 int64_t downscale_factor) { in cpu_pixel_unshuffle_channels_last()
223 int64_t downscale_factor) { in pixel_unshuffle_kernel_impl()
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_pixel_unshuffle.cpp | 19 int64_t downscale_factor, in pixel_unshuffle_impl()
73 int64_t downscale_factor, in pixel_unshuffle_out()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | PixelShuffle.cpp | 47 Tensor pixel_unshuffle_cpu(const Tensor& self, int64_t downscale_factor) { in pixel_unshuffle_cpu()
116 Tensor math_pixel_unshuffle(const Tensor& self, int64_t downscale_factor) { in math_pixel_unshuffle()
|
| H A D | PixelShuffle.h | 21 inline void check_pixel_unshuffle_shapes(const Tensor& self, int64_t downscale_factor) { in check_pixel_unshuffle_shapes()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | pixelshuffle.h | 15 inline Tensor pixel_unshuffle(const Tensor& input, int64_t downscale_factor) { in pixel_unshuffle()
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | copy_ops_util.cpp | 332 int64_t downscale_factor, in check_pixel_unshuffle_args()
367 int64_t downscale_factor, in get_pixel_unshuffle_out_target_size()
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/ |
| H A D | ts_native_functions.cpp | 435 int64_t downscale_factor) { in pixel_unshuffle()
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset9.py | 4198 def pixel_unshuffle(g: jit_utils.GraphContext, self, downscale_factor): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_mps.py | 1355 downscale_factor=None): argument
|
| H A D | test_nn.py | 4443 downscale_factor=None): argument
|