| /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() function
12 return torch::pixel_shuffle(input, upscale_factor); in pixel_shuffle()
22 /// https://pytorch.org/docs/main/nn.functional.html#torch.nn.functional.pixel_shuffle
31 /// F::pixel_shuffle(x, F::PixelShuffleFuncOptions(2));
33 inline Tensor pixel_shuffle( in pixel_shuffle() function
36 return detail::pixel_shuffle(input, options.upscale_factor()); in pixel_shuffle()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | PixelShuffle.h | 9 "pixel_shuffle expects input to have at least 3 dimensions, but got input with ", in check_pixel_shuffle_shapes()
12 "pixel_shuffle expects a positive upscale_factor, but got ", in check_pixel_shuffle_shapes()
17 … "pixel_shuffle expects its input's 'channel' dimension to be divisible by the square of " in check_pixel_shuffle_shapes()
|
| H A D | ts_native_functions.yaml | 200 - pixel_shuffle
|
| H A D | PixelShuffle.cpp | 112 // pixel_shuffle expects to *never* return an alias of the input. in math_pixel_shuffle()
|
| /aosp_15_r20/external/pytorch/torch/nn/modules/ |
| H A D | pixelshuffle.py | 41 >>> pixel_shuffle = nn.PixelShuffle(3)
43 >>> output = pixel_shuffle(input)
59 return F.pixel_shuffle(input, self.upscale_factor)
|
| /aosp_15_r20/external/executorch/backends/apple/mps/runtime/operations/ |
| H A D | ShapeOps.mm | 171 ndims >= 3, Internal, "pixel_shuffle requires tensor with at least 3 dimensions.");
179 "pixel_shuffle channels must be divisible by upscale factor squared.");
187 name:@"pixel_shuffle"];
|
| /aosp_15_r20/external/pytorch/test/onnx/model_defs/ |
| H A D | super_resolution.py | 14 self.pixel_shuffle = nn.PixelShuffle(upscale_factor)
22 x = self.pixel_shuffle(self.conv4(x))
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/options/ |
| H A D | pixelshuffle.h | 39 /// Options for `torch::nn::functional::pixel_shuffle`.
47 /// F::pixel_shuffle(x, F::PixelShuffleFuncOptions(2));
|
| /aosp_15_r20/external/pytorch/torch/ao/ns/fx/ |
| H A D | mappings.py | 341 F.pixel_shuffle,
554 F.pixel_shuffle,
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset11.py | 66 "pixel_shuffle",
325 @_onnx_symbolic("aten::pixel_shuffle")
327 def pixel_shuffle(g: jit_utils.GraphContext, self, upscale_factor): function
330 return symbolic_helper._unimplemented("pixel_shuffle", "only support 4d input")
|
| /aosp_15_r20/external/pytorch/torch/_refs/nn/functional/ |
| H A D | __init__.py | 1218 @register_decomposition(aten.pixel_shuffle)
1220 def pixel_shuffle(self: Tensor, upscale_factor: int): function
1223 …lambda: f"pixel_shuffle expects input to have at least 3 dimensions, but got input with {self.dim}…
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/ |
| H A D | ts_native_functions.cpp | 427 at::Tensor LazyNativeFunctions::pixel_shuffle( in pixel_shuffle() function in torch::lazy::LazyNativeFunctions
431 pixel_shuffle)>::call(self, upscale_factor); in pixel_shuffle()
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | top_operators_github_usage.py | 308 ("nn.functional.pixel_shuffle", 209),
451 ("nn.PixelShuffle", 6571, "nn.functional.pixel_shuffle"),
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/ |
| H A D | pixelshuffle.cpp | 19 return F::detail::pixel_shuffle(input, options.upscale_factor()); in forward()
|
| /aosp_15_r20/external/executorch/backends/qualcomm/builders/ |
| H A D | op_depth_to_space.py | 21 target = ["aten.pixel_shuffle.default"]
|
| /aosp_15_r20/external/executorch/backends/mediatek/ |
| H A D | partitioner.py | 74 torch.ops.aten.pixel_shuffle.default,
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | nn.functional.rst | 200 pixel_shuffle
|
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_models.py | 469 self.pixel_shuffle = nn.PixelShuffle(upscale_factor)
475 x = self.pixel_shuffle(self.conv4(x))
|
| /aosp_15_r20/external/executorch/backends/qualcomm/tests/ |
| H A D | models.py | 707 self.pixel_shuffle = torch.nn.PixelShuffle(scale)
710 return self.pixel_shuffle(x)
|
| /aosp_15_r20/external/executorch/kernels/test/ |
| H A D | op_pixel_shuffle_test.cpp | 42 // Destination for the pixel_shuffle. in test_pixel_shuffle()
|
| /aosp_15_r20/external/pytorch/functorch/op_analysis/ |
| H A D | public_api | 496 nn.functional.pixel_shuffle
|
| /aosp_15_r20/external/executorch/kernels/aten/ |
| H A D | functions.yaml | 292 - op: pixel_shuffle.out
|
| /aosp_15_r20/external/executorch/backends/apple/mps/operators/ |
| H A D | shape_ops.py | 167 target = "aten.pixel_shuffle.default"
|
| /aosp_15_r20/external/pytorch/test/edge/ |
| H A D | selected_operators.yaml | 296 aten::pixel_shuffle.out:
|
| /aosp_15_r20/external/pytorch/functorch/dim/ |
| H A D | README.md | 24 def pixel_shuffle(img: torch.Tensor, upscale_factor=2):
396 def pixel_shuffle(img, upscale_factor=2):
|