Home
last modified time | relevance | path

Searched full:upsample_linear1d (Results 1 – 25 of 28) sorted by relevance

12

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DUpSampleLinear1d.cpp14 #include <ATen/ops/upsample_linear1d.h>
22 TORCH_META_FUNC(upsample_linear1d) ( in TORCH_META_FUNC() argument
91 Tensor upsample_linear1d( in upsample_linear1d() function
98 return at::upsample_linear1d(input, osize, align_corners, scale_w); in upsample_linear1d()
H A Dnative_functions.yaml12487 - func: upsample_linear1d.vec(Tensor input, SymInt[]? output_size, bool align_corners, float[]? sca…
12489 autogen: upsample_linear1d.vec_out
12538 - func: upsample_linear1d.out(Tensor self, SymInt[1] output_size, bool align_corners, float? scales…
12546 - func: upsample_linear1d(Tensor self, SymInt[1] output_size, bool align_corners, float? scales=Non…
12548 structured_delegate: upsample_linear1d.out
/aosp_15_r20/external/pytorch/torch/onnx/_internal/exporter/
H A D_decomp.py57 aten.upsample_linear1d.default,
58 aten.upsample_linear1d.vec,
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesDecompositions.cpp343 m.impl("upsample_linear1d.vec", native::upsample_linear1d); in TORCH_LIBRARY_IMPL()
H A DBatchRulesModules.cpp386 EXISTING_BDIM(upsample_linear1d); in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset8.py82 "aten::upsample_linear1d",
83 decorate=[symbolic_helper._apply_params("upsample_linear1d", 3, "linear")],
H A Dsymbolic_opset10.py443 "aten::upsample_linear1d",
444 decorate=[symbolic_helper._apply_params("upsample_linear1d", 3, "linear")],
H A Dsymbolic_opset11.py347 "aten::upsample_linear1d",
348 decorate=[symbolic_helper._apply_params("upsample_linear1d", 3, "linear")],
H A Dsymbolic_opset9.py1871 "aten::upsample_linear1d",
1873 symbolic_helper._apply_params("upsample_linear1d", 3, "linear"),
1874 _export("upsample_linear1d"),
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/
H A Dquantization_patterns.h851 "aten::upsample_linear1d", in quant_fusion_pattern_and_replacements()
872 auto upsample_linear1d = getInputTensorQParamOpFusionInfo( in quant_fusion_pattern_and_replacements() local
873 "aten::upsample_linear1d", {"%output_size", "%align_corners", "%scales"}); in quant_fusion_pattern_and_replacements()
1059 std::move(upsample_linear1d), in quant_fusion_pattern_and_replacements()
H A Dhelper.cpp155 "upsample_linear1d",
/aosp_15_r20/external/pytorch/test/mobile/lightweight_dispatch/
H A Dtests_setup.py74 # upsample_linear1d.vec(Tensor input, int[]? output_size, bool align_corners, float[]? scale_factor…
/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dcoverage.yaml621 - aten::upsample_linear1d.vec
1003 aten::upsample_linear1d.vec: 1
H A Dmodel_ops.yaml380 aten::upsample_linear1d.vec: 1
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/utils/
H A Dop_registry.cpp49 …"aten::upsample_linear1d(Tensor self, int[] output_size, bool align_corners, float? scales) -> Ten… in nn_ops_first_input_preserving()
/aosp_15_r20/external/pytorch/test/export/
H A Dtesting.py97 aten.upsample_linear1d.vec,
/aosp_15_r20/external/pytorch/functorch/op_analysis/
H A Dannotated_ops477 upsample_linear1d, misc
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_dtype_analysis.py80 "upsample_linear1d",
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/
H A DUpSample.mm23 #include <ATen/ops/upsample_linear1d.h>
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Dupsampling.h185 return torch::upsample_linear1d( in interpolate()
/aosp_15_r20/external/pytorch/torch/_decomp/
H A D__init__.py470 aten.upsample_linear1d,
H A Ddecompositions.py3629 @aten.upsample_linear1d.vec.py_impl(DispatchKey.CompositeImplicitAutograd)
3630 @aten.upsample_linear1d.vec.py_impl(DispatchKey.Autograd)
3641 @register_decomposition([aten.upsample_linear1d.default, aten.upsample_linear1d.out])
3643 def upsample_linear1d( function
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dregister_prim_ops_fulljit.cpp537 return at::upsample_linear1d( in interpolate()
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A Dautocast_mode.h863 _(upsample_linear1d) \
/aosp_15_r20/external/pytorch/docs/source/
H A Dconf.py1719 "upsample_linear1d",

12