| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | UpSampleNearest1d.cpp | 15 #include <ATen/ops/upsample_nearest1d.h>
23 TORCH_META_FUNC(upsample_nearest1d) ( in TORCH_META_FUNC() argument
125 Tensor upsample_nearest1d( in upsample_nearest1d() function
131 return at::upsample_nearest1d(input, osize, scale_w); in upsample_nearest1d()
|
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/exporter/ |
| H A D | _decomp.py | 59 aten.upsample_nearest1d.default,
60 aten.upsample_nearest1d.vec,
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | UpSampleNearest1d.cu | 123 // upsample_nearest1d meta call makes sure `nbatch != 0` in upsample_nearest1d_out_cuda_template()
171 // upsample_nearest1d meta call makes sure `nbatch != 0` in upsample_nearest1d_backward_out_cuda_template()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesDecompositions.cpp | 344 m.impl("upsample_nearest1d.vec", native::upsample_nearest1d); in TORCH_LIBRARY_IMPL()
|
| H A D | BatchRulesModules.cpp | 387 EXISTING_BDIM(upsample_nearest1d); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset8.py | 70 "aten::upsample_nearest1d",
71 decorate=[symbolic_helper._apply_params("upsample_nearest1d", 3, "nearest")],
|
| H A D | symbolic_opset10.py | 431 "aten::upsample_nearest1d",
432 decorate=[symbolic_helper._apply_params("upsample_nearest1d", 3, "nearest")],
|
| H A D | symbolic_opset11.py | 335 "aten::upsample_nearest1d",
336 decorate=[symbolic_helper._apply_params("upsample_nearest1d", 3, "nearest")],
|
| H A D | symbolic_opset9.py | 1850 "aten::upsample_nearest1d",
1852 symbolic_helper._apply_params("upsample_nearest1d", 3, "nearest"),
1853 _export("upsample_nearest1d"),
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/ |
| H A D | quantization_patterns.h | 842 "aten::upsample_nearest1d", {"%output_size", "%scale_factors"}); in quant_fusion_pattern_and_replacements()
862 auto upsample_nearest1d = getInputTensorQParamOpFusionInfo( in quant_fusion_pattern_and_replacements() local
863 "aten::upsample_nearest1d", {"%output_size", "%scales"}); in quant_fusion_pattern_and_replacements()
1056 std::move(upsample_nearest1d), in quant_fusion_pattern_and_replacements()
|
| H A D | helper.cpp | 152 "upsample_nearest1d",
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | decompositions.py | 2764 @register_decomposition(aten.upsample_nearest1d.vec)
2767 @aten.upsample_nearest1d.vec.py_impl(DispatchKey.CompositeImplicitAutograd)
2768 @aten.upsample_nearest1d.vec.py_impl(DispatchKey.Autograd)
2838 @register_decomposition([aten.upsample_nearest1d.default, aten.upsample_nearest1d.out])
2839 @aten.upsample_nearest1d.default.py_impl(DispatchKey.CompositeImplicitAutograd)
2840 @aten.upsample_nearest1d.default.py_impl(DispatchKey.Autograd)
2842 def upsample_nearest1d( function
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | coverage.yaml | 622 - aten::upsample_nearest1d.vec
1004 aten::upsample_nearest1d.vec: 2
|
| H A D | model_ops.yaml | 381 aten::upsample_nearest1d.vec: 2
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/utils/ |
| H A D | op_registry.cpp | 50 "aten::upsample_nearest1d(Tensor self, int[] output_size, float? scales) -> Tensor", in nn_ops_first_input_preserving()
|
| /aosp_15_r20/external/pytorch/test/export/ |
| H A D | testing.py | 79 aten.upsample_nearest1d.vec,
|
| /aosp_15_r20/external/pytorch/functorch/op_analysis/ |
| H A D | annotated_ops | 481 upsample_nearest1d, misc
|
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_dtype_analysis.py | 81 "upsample_nearest1d",
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/ |
| H A D | UpSample.mm | 27 #include <ATen/ops/upsample_nearest1d.h>
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | upsampling.h | 138 return torch::upsample_nearest1d( in interpolate()
|
| /aosp_15_r20/external/pytorch/test/expect/ |
| H A D | HasDecompTest.test_aten_core_operators.expect | 516 aten::upsample_nearest1d.out
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | register_prim_ops_fulljit.cpp | 510 return at::upsample_nearest1d( in interpolate()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | autocast_mode.h | 857 _(upsample_nearest1d) \
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_torchinductor.py | 5730 aten.upsample_nearest1d(a, [74], None),
5731 aten.upsample_nearest1d(a, [70], None),
5732 aten.upsample_nearest1d(a, [45], None),
5733 aten.upsample_nearest1d(a, [36], None),
5734 aten.upsample_nearest1d(a, None, [2.0]),
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | conf.py | 1720 "upsample_nearest1d",
|