Searched full:replication_pad1d (Results 1 – 25 of 26) sorted by relevance
12
| /external/executorch/kernels/portable/cpu/ |
| D | op_replication_pad1d.cpp | 39 constexpr auto name = "replication_pad1d.out"; in replication_pad1d_out()
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | ReplicationPadding.cu | 423 "replication_pad1d only supports input tensors with less than 2^63 - 1 elements"); in TORCH_IMPL_FUNC()
484 "replication_pad1d only supports input tensors with less than 2^63 - 1 elements"); in TORCH_IMPL_FUNC()
486 "replication_pad1d only supports output tensors with less than 2^63 - 1 elements"); in TORCH_IMPL_FUNC()
|
| /external/executorch/kernels/aten/ |
| D | functions.yaml | 322 - op: replication_pad1d.out
|
| /external/pytorch/torch/csrc/jit/passes/utils/ |
| D | op_registry.cpp | 45 "aten::replication_pad1d(Tensor self, int[] padding) -> Tensor", in nn_ops_first_input_preserving()
|
| /external/pytorch/functorch/op_analysis/ |
| D | annotated_ops | 474 replication_pad1d, misc
|
| /external/pytorch/aten/src/ATen/native/ |
| D | ReplicationPadding.cpp | 25 TORCH_META_FUNC(replication_pad1d) ( in TORCH_META_FUNC() argument
|
| D | PadNd.cpp | 22 #include <ATen/ops/replication_pad1d.h>
|
| D | native_functions.yaml | 12393 - func: replication_pad1d.out(Tensor self, SymInt[2] padding, *, Tensor(a!) out) -> Tensor(a!)
12401 - func: replication_pad1d(Tensor self, SymInt[2] padding) -> Tensor
12403 structured_delegate: replication_pad1d.out
|
| /external/pytorch/test/jit/ |
| D | test_dtype_analysis.py | 76 "replication_pad1d",
|
| /external/pytorch/torch/_decomp/ |
| D | __init__.py | 401 aten.replication_pad1d,
|
| D | decompositions.py | 4570 @register_decomposition(aten.replication_pad1d)
|
| /external/pytorch/docs/source/ |
| D | amp.rst | 431 ``replication_pad1d``,
|
| /external/pytorch/test/mobile/model_test/ |
| D | model_ops.yaml | 304 aten::replication_pad1d: 1
|
| D | coverage.yaml | 510 - aten::replication_pad1d
941 aten::replication_pad1d: 1
|
| /external/pytorch/aten/src/ATen/ |
| D | autocast_mode.cpp | 386 KERNEL_CPU(replication_pad1d, fp32) in TORCH_LIBRARY_IMPL()
|
| /external/executorch/kernels/portable/ |
| D | functions.yaml | 730 - op: replication_pad1d.out
|
| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchRulesModules.cpp | 372 EXISTING_BDIM(replication_pad1d); in TORCH_LIBRARY_IMPL()
|
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | PaddingKernel.cpp | 607 "replication_pad1d", [&] { in replication_pad1d_kernel_impl()
|
| /external/pytorch/torch/onnx/ |
| D | symbolic_opset11.py | 732 @_onnx_symbolic("aten::replication_pad1d")
|
| D | symbolic_opset9.py | 1816 @_onnx_symbolic("aten::replication_pad1d")
|
| /external/pytorch/tools/autograd/ |
| D | gen_variable_type.py | 325 "replication_pad1d",
|
| D | derivatives.yaml | 2183 - name: replication_pad1d(Tensor self, SymInt[2] padding) -> Tensor
|
| /external/pytorch/torch/_dynamo/ |
| D | trace_rules.py | 1054 "torch._C._nn.replication_pad1d",
|
| /external/pytorch/test/cpp/lazy/ |
| D | test_lazy_ops.cpp | 9719 torch::Tensor output = torch::replication_pad1d(input, pad); in TEST_F()
9722 torch::Tensor lazy_output = torch::replication_pad1d(lazy_input, pad); in TEST_F()
9731 torch::Tensor output = torch::replication_pad1d(input, pad); in TEST_F()
9734 torch::Tensor lazy_output = torch::replication_pad1d(lazy_input, pad); in TEST_F()
9742 return torch::replication_pad1d(inputs[0], pad); in TEST_F()
|
| /external/pytorch/torch/ |
| D | _meta_registrations.py | 1714 @register_meta(aten.replication_pad1d)
|
12