| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Padding.cpp | 101 Tensor replication_pad2d(const Tensor& self_arg, IntArrayRef padding) { in replication_pad2d() function
102 return pad2d(self_arg, padding, VK_KERNEL(replication_pad2d)); in replication_pad2d()
112 TORCH_SELECTIVE_NAME("aten::replication_pad2d"), in TORCH_LIBRARY_IMPL()
113 TORCH_FN(replication_pad2d)); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | __init__.py | 402 aten.replication_pad2d,
406 aten.replication_pad2d,
|
| H A D | decompositions.py | 4571 @register_decomposition(aten.replication_pad2d)
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_replication_pad2d.cpp | 39 constexpr auto name = "replication_pad2d.out"; in replication_pad2d_out()
|
| /aosp_15_r20/external/executorch/kernels/aten/ |
| H A D | functions.yaml | 324 - op: replication_pad2d.out
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | coverage.yaml | 511 - aten::replication_pad2d
942 aten::replication_pad2d: 2
|
| H A D | model_ops.yaml | 305 aten::replication_pad2d: 1
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/utils/ |
| H A D | op_registry.cpp | 46 "aten::replication_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 | 475 replication_pad2d, misc
|
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_dtype_analysis.py | 77 "replication_pad2d",
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | ReplicationPadding.cpp | 86 TORCH_META_FUNC(replication_pad2d) ( in TORCH_META_FUNC() argument
|
| H A D | PadNd.cpp | 23 #include <ATen/ops/replication_pad2d.h>
|
| H A D | native_functions.yaml | 12417 - func: replication_pad2d.out(Tensor self, SymInt[4] padding, *, Tensor(a!) out) -> Tensor(a!)
12425 - func: replication_pad2d(Tensor self, SymInt[4] padding) -> Tensor
12427 structured_delegate: replication_pad2d.out
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | amp.rst | 432 ``replication_pad2d``,
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | autocast_mode.cpp | 387 KERNEL_CPU(replication_pad2d, fp32) in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/executorch/kernels/portable/ |
| H A D | functions.yaml | 735 - op: replication_pad2d.out
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesModules.cpp | 373 EXISTING_BDIM(replication_pad2d); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | PaddingKernel.cpp | 626 "replication_pad2d", [&] { in replication_pad2d_kernel_impl()
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset11.py | 733 @_onnx_symbolic("aten::replication_pad2d")
|
| /aosp_15_r20/external/pytorch/tools/autograd/ |
| H A D | gen_variable_type.py | 326 "replication_pad2d",
|
| H A D | derivatives.yaml | 2187 - name: replication_pad2d(Tensor self, SymInt[4] padding) -> Tensor
|
| /aosp_15_r20/external/pytorch/test/cpp/lazy/ |
| H A D | test_lazy_ops.cpp | 9760 torch::Tensor output = torch::replication_pad2d(input, pad); in TEST_F()
9763 torch::Tensor lazy_output = torch::replication_pad2d(lazy_input, pad); in TEST_F()
9772 torch::Tensor output = torch::replication_pad2d(input, pad); in TEST_F()
9775 torch::Tensor lazy_output = torch::replication_pad2d(lazy_input, pad); in TEST_F()
9783 return torch::replication_pad2d(inputs[0], pad); in TEST_F()
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | trace_rules.py | 1055 "torch._C._nn.replication_pad2d",
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | vulkan_api_test.cpp | 4279 TEST_F(VulkanAPITest, replication_pad2d) { in TEST_F() argument
4285 const auto out_cpu = at::replication_pad2d(a_cpu, padding_params); in TEST_F()
4286 const auto out_vulkan = at::replication_pad2d(a_vulkan, padding_params).cpu(); in TEST_F()
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 1821 @register_meta(aten.replication_pad2d)
|