| /aosp_15_r20/external/pytorch/aten/src/ATen/native/metal/ops/ |
| H A D | MetalPadding.mm | 15 static Tensor reflection_pad2d(const Tensor& input, IntArrayRef padding) {
57 specializedPipelineState:"reflection_pad2d"
87 m.impl(TORCH_SELECTIVE_NAME("aten::reflection_pad2d"), TORCH_FN(reflection_pad2d));
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Padding.cpp | 97 Tensor reflection_pad2d(const Tensor& self_arg, IntArrayRef padding) { in reflection_pad2d() function
98 return pad2d(self_arg, padding, VK_KERNEL(reflection_pad2d)); in reflection_pad2d()
109 TORCH_SELECTIVE_NAME("aten::reflection_pad2d"), in TORCH_LIBRARY_IMPL()
110 TORCH_FN(reflection_pad2d)); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_reflection_pad2d.cpp | 47 constexpr auto name = "reflection_pad2d.out"; in reflection_pad2d_out()
|
| /aosp_15_r20/external/executorch/kernels/aten/ |
| H A D | functions.yaml | 318 - op: reflection_pad2d.out
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | coverage.yaml | 500 - aten::reflection_pad2d
934 aten::reflection_pad2d: 1
|
| H A D | model_ops.yaml | 296 aten::reflection_pad2d: 2
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/utils/ |
| H A D | op_registry.cpp | 43 "aten::reflection_pad2d(Tensor self, int[] padding) -> Tensor", in nn_ops_first_input_preserving()
|
| /aosp_15_r20/external/pytorch/functorch/op_analysis/ |
| H A D | annotated_ops | 473 reflection_pad2d, misc
|
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_dtype_analysis.py | 74 "reflection_pad2d",
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | PadNd.cpp | 20 #include <ATen/ops/reflection_pad2d.h>
|
| /aosp_15_r20/external/pytorch/benchmarks/dynamo/microbenchmarks/operator_inp_logs/torchbench_train/ |
| H A D | pytorch_CycleGAN_and_pix2pix_training.txt | 47 Operator: aten.reflection_pad2d.default
|
| H A D | Background_Matting_training.txt | 79 Operator: aten.reflection_pad2d.default
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | __init__.py | 397 aten.reflection_pad2d,
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | amp.rst | 430 ``reflection_pad2d``,
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/metal/mpscnn/tests/ |
| H A D | MPSCNNTests.mm | 901 auto Y1 = at::reflection_pad2d(X1, {2,4,7,9});
902 auto Y2 = at::reflection_pad2d(X2, {2,4,7,9}).cpu();
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | autocast_mode.cpp | 385 KERNEL_CPU(reflection_pad2d, fp32) in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/executorch/kernels/portable/ |
| H A D | functions.yaml | 720 - op: reflection_pad2d.out
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesModules.cpp | 370 EXISTING_BDIM(reflection_pad2d); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | PaddingKernel.cpp | 517 "reflection_pad2d", [&] { in reflection_pad2d_kernel_impl()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/metal/ |
| H A D | MetalShaders.h | 484 kernel void reflection_pad2d(texture2d_array<half, access::read> in_arr[[texture(0), function_const…
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset11.py | 724 @_onnx_symbolic("aten::reflection_pad2d")
|
| /aosp_15_r20/external/pytorch/tools/autograd/ |
| H A D | gen_variable_type.py | 311 "reflection_pad2d",
|
| /aosp_15_r20/external/pytorch/test/cpp/lazy/ |
| H A D | test_lazy_ops.cpp | 9677 torch::Tensor output = torch::reflection_pad2d(input, pad); in TEST_F()
9680 torch::Tensor lazy_output = torch::reflection_pad2d(lazy_input, pad); in TEST_F()
9690 torch::Tensor output = torch::reflection_pad2d(input, pad); in TEST_F()
9693 torch::Tensor lazy_output = torch::reflection_pad2d(lazy_input, pad); in TEST_F()
9701 return torch::reflection_pad2d(inputs[0], pad); in TEST_F()
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | trace_rules.py | 1050 "torch._C._nn.reflection_pad2d",
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | vulkan_api_test.cpp | 4186 TEST_F(VulkanAPITest, reflection_pad2d) { in TEST_F() argument
4190 const auto out_cpu = at::reflection_pad2d(a_cpu, {9,8,5,12}); in TEST_F()
4191 const auto out_vulkan = at::reflection_pad2d(a_vulkan, {9,8,5,12}).cpu(); in TEST_F()
|