Home
last modified time | relevance | path

Searched full:replication_pad3d (Results 1 – 24 of 24) sorted by relevance

/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_replication_pad3d.cpp39 constexpr auto name = "replication_pad3d.out"; in replication_pad3d_out()
/aosp_15_r20/external/executorch/kernels/aten/
H A Dfunctions.yaml326 - op: replication_pad3d.out
/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dcoverage.yaml512 - aten::replication_pad3d
943 aten::replication_pad3d: 1
H A Dmodel_ops.yaml306 aten::replication_pad3d: 1
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/utils/
H A Dop_registry.cpp47 "aten::replication_pad3d(Tensor self, int[] padding) -> Tensor", in nn_ops_first_input_preserving()
/aosp_15_r20/external/pytorch/functorch/op_analysis/
H A Dannotated_ops476 replication_pad3d, misc
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_dtype_analysis.py78 "replication_pad3d",
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DReplicationPadding.cpp126 TORCH_META_FUNC(replication_pad3d) ( in TORCH_META_FUNC() argument
H A DPadNd.cpp24 #include <ATen/ops/replication_pad3d.h>
H A Dnative_functions.yaml12444 - func: replication_pad3d.out(Tensor self, SymInt[6] padding, *, Tensor(a!) out) -> Tensor(a!)
12452 - func: replication_pad3d(Tensor self, SymInt[6] padding) -> Tensor
12454 structured_delegate: replication_pad3d.out
/aosp_15_r20/external/pytorch/torch/_decomp/
H A D__init__.py403 aten.replication_pad3d,
H A Ddecompositions.py4572 @register_decomposition(aten.replication_pad3d)
/aosp_15_r20/external/pytorch/docs/source/
H A Damp.rst433 ``replication_pad3d``,
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A Dautocast_mode.cpp388 KERNEL_CPU(replication_pad3d, fp32) in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/executorch/kernels/portable/
H A Dfunctions.yaml740 - op: replication_pad3d.out
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesModules.cpp374 EXISTING_BDIM(replication_pad3d); in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DPaddingKernel.cpp671 "replication_pad3d", [&] { in replication_pad3d_kernel_impl()
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset11.py734 @_onnx_symbolic("aten::replication_pad3d")
H A Dsymbolic_opset9.py1818 @_onnx_symbolic("aten::replication_pad3d")
/aosp_15_r20/external/pytorch/tools/autograd/
H A Dgen_variable_type.py327 "replication_pad3d",
H A Dderivatives.yaml2191 - name: replication_pad3d(Tensor self, SymInt[6] padding) -> Tensor
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dtrace_rules.py1056 "torch._C._nn.replication_pad3d",
/aosp_15_r20/external/pytorch/torch/
H A D_meta_registrations.py1937 @register_meta(aten.replication_pad3d)
/aosp_15_r20/external/pytorch/test/
H A Dtest_nn.py8585 torch._C._nn.replication_pad3d(torch.randn([2]), padding=[])