Searched full:replication_pad3d (Results 1 – 24 of 24) sorted by relevance
| /external/executorch/kernels/portable/cpu/ |
| D | op_replication_pad3d.cpp | 39 constexpr auto name = "replication_pad3d.out"; in replication_pad3d_out()
|
| /external/executorch/kernels/aten/ |
| D | functions.yaml | 326 - op: replication_pad3d.out
|
| /external/pytorch/torch/csrc/jit/passes/utils/ |
| D | op_registry.cpp | 47 "aten::replication_pad3d(Tensor self, int[] padding) -> Tensor", in nn_ops_first_input_preserving()
|
| /external/pytorch/functorch/op_analysis/ |
| D | annotated_ops | 476 replication_pad3d, misc
|
| /external/pytorch/aten/src/ATen/native/ |
| D | ReplicationPadding.cpp | 126 TORCH_META_FUNC(replication_pad3d) ( in TORCH_META_FUNC() argument
|
| D | PadNd.cpp | 24 #include <ATen/ops/replication_pad3d.h>
|
| D | native_functions.yaml | 12444 - 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
|
| /external/pytorch/test/jit/ |
| D | test_dtype_analysis.py | 78 "replication_pad3d",
|
| /external/pytorch/torch/_decomp/ |
| D | __init__.py | 403 aten.replication_pad3d,
|
| D | decompositions.py | 4572 @register_decomposition(aten.replication_pad3d)
|
| /external/pytorch/docs/source/ |
| D | amp.rst | 433 ``replication_pad3d``,
|
| /external/pytorch/test/mobile/model_test/ |
| D | model_ops.yaml | 306 aten::replication_pad3d: 1
|
| D | coverage.yaml | 512 - aten::replication_pad3d
943 aten::replication_pad3d: 1
|
| /external/pytorch/aten/src/ATen/ |
| D | autocast_mode.cpp | 388 KERNEL_CPU(replication_pad3d, fp32) in TORCH_LIBRARY_IMPL()
|
| /external/executorch/kernels/portable/ |
| D | functions.yaml | 740 - op: replication_pad3d.out
|
| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchRulesModules.cpp | 374 EXISTING_BDIM(replication_pad3d); in TORCH_LIBRARY_IMPL()
|
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | PaddingKernel.cpp | 671 "replication_pad3d", [&] { in replication_pad3d_kernel_impl()
|
| /external/pytorch/torch/onnx/ |
| D | symbolic_opset11.py | 734 @_onnx_symbolic("aten::replication_pad3d")
|
| D | symbolic_opset9.py | 1818 @_onnx_symbolic("aten::replication_pad3d")
|
| /external/pytorch/tools/autograd/ |
| D | gen_variable_type.py | 327 "replication_pad3d",
|
| D | derivatives.yaml | 2191 - name: replication_pad3d(Tensor self, SymInt[6] padding) -> Tensor
|
| /external/pytorch/torch/_dynamo/ |
| D | trace_rules.py | 1056 "torch._C._nn.replication_pad3d",
|
| /external/pytorch/torch/ |
| D | _meta_registrations.py | 1937 @register_meta(aten.replication_pad3d)
|
| /external/pytorch/test/ |
| D | test_nn.py | 8585 torch._C._nn.replication_pad3d(torch.randn([2]), padding=[])
|