| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | PixelShuffleKernel.cpp | 19 int64_t upscale_factor) { in cpu_pixel_shuffle()
59 int64_t upscale_factor) { in cpu_pixel_shuffle_channels_last()
199 int64_t upscale_factor) { in pixel_shuffle_kernel_impl()
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_pixel_shuffle.cpp | 17 void pixel_shuffle_impl(const Tensor& in, int64_t upscale_factor, Tensor& out) { in pixel_shuffle_impl()
70 int64_t upscale_factor, in pixel_shuffle_out()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | PixelShuffle.cpp | 23 Tensor pixel_shuffle_cpu(const Tensor& self, int64_t upscale_factor) { in pixel_shuffle_cpu()
75 Tensor math_pixel_shuffle(const Tensor& self, int64_t upscale_factor) { in math_pixel_shuffle()
|
| H A D | PixelShuffle.h | 7 inline void check_pixel_shuffle_shapes(const Tensor& self, int64_t upscale_factor) { in check_pixel_shuffle_shapes()
|
| /aosp_15_r20/external/pytorch/test/onnx/model_defs/ |
| H A D | super_resolution.py | 6 def __init__(self, upscale_factor): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | pixelshuffle.h | 11 inline Tensor pixel_shuffle(const Tensor& input, int64_t upscale_factor) { in pixel_shuffle()
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | copy_ops_util.cpp | 319 int64_t upscale_factor, in check_pixel_shuffle_args()
345 int64_t upscale_factor, in get_pixel_shuffle_out_target_size()
|
| /aosp_15_r20/external/executorch/kernels/test/ |
| H A D | op_pixel_shuffle_test.cpp | 29 int64_t upscale_factor, in op_pixel_shuffle_out()
|
| H A D | op_pixel_unshuffle_test.cpp | 29 int64_t upscale_factor, in op_pixel_unshuffle_out()
|
| /aosp_15_r20/external/executorch/backends/apple/mps/serialization/ |
| H A D | mps_graph_schema.py | 538 upscale_factor: int = 1 variable in MPSPixelShuffle
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/ |
| H A D | ts_native_functions.cpp | 429 int64_t upscale_factor) { in pixel_shuffle()
|
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_models.py | 461 def __init__(self, upscale_factor): argument
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset11.py | 327 def pixel_shuffle(g: jit_utils.GraphContext, self, upscale_factor): argument
|
| H A D | symbolic_opset9.py | 4126 def pixel_shuffle(g: jit_utils.GraphContext, self, upscale_factor): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_mps.py | 1295 upscale_factor=None, is_contiguous=True): argument
1326 def _verify_pixel_shuffle(input, output, upscale_factor): argument
|
| H A D | test_nn.py | 4401 upscale_factor=None): argument
4404 def _verify_pixel_shuffle(input, output, upscale_factor): argument
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 6128 def meta_pixel_shuffle(self, upscale_factor): argument
|
| /aosp_15_r20/external/pytorch/test/cpp/lazy/ |
| H A D | test_lazy_ops.cpp | 8747 int upscale_factor = 3; in TEST_F() local
|