| /aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/ |
| H A D | decomposition_skip.py | 88 op_callable = torch._C._nn.upsample_bilinear2d # type: ignore[attr-defined]
90 new_op_name = "upsample_bilinear2d"
99 …torch._C._nn.upsample_bilinear2d = torch.ops.onnx_export.upsample_bilinear2d # type: ignore[attr-…
111 torch._C._nn.upsample_bilinear2d = cls.op_callable # type: ignore[attr-defined]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | UpSampleBilinear2d.cpp | 18 #include <ATen/ops/upsample_bilinear2d.h>
26 TORCH_META_FUNC(upsample_bilinear2d) ( in TORCH_META_FUNC() argument
160 Tensor upsample_bilinear2d( in upsample_bilinear2d() function
168 return at::upsample_bilinear2d(input, osize, align_corners, scale_h, scale_w); in upsample_bilinear2d()
|
| H A D | ts_native_functions.yaml | 139 - upsample_bilinear2d
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Upsample.cpp | 97 static Tensor upsample_bilinear2d( in upsample_bilinear2d() function
186 TORCH_SELECTIVE_NAME("aten::upsample_bilinear2d"), in TORCH_LIBRARY_IMPL()
187 TORCH_FN(upsample_bilinear2d)); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/executorch/backends/mediatek/ |
| H A D | partitioner.py | 75 torch.ops.aten.upsample_bilinear2d.default,
76 torch.ops.aten.upsample_bilinear2d.vec,
|
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/exporter/ |
| H A D | _decomp.py | 55 aten.upsample_bilinear2d.default,
56 aten.upsample_bilinear2d.vec,
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesDecompositions.cpp | 339 m.impl("upsample_bilinear2d.vec", native::upsample_bilinear2d); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/executorch/backends/xnnpack/partition/config/ |
| H A D | generic_node_configs.py | 304 target_name = "upsample_bilinear2d.vec"
310 return torch.ops.aten.upsample_bilinear2d.vec
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset8.py | 86 "aten::upsample_bilinear2d",
87 decorate=[symbolic_helper._apply_params("upsample_bilinear2d", 4, "linear")],
|
| H A D | symbolic_opset10.py | 447 "aten::upsample_bilinear2d",
448 decorate=[symbolic_helper._apply_params("upsample_bilinear2d", 4, "linear")],
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/ |
| H A D | quantization_patterns.h | 855 "aten::upsample_bilinear2d", in quant_fusion_pattern_and_replacements()
875 auto upsample_bilinear2d = getInputTensorQParamOpFusionInfo( in quant_fusion_pattern_and_replacements() local
876 "aten::upsample_bilinear2d", in quant_fusion_pattern_and_replacements()
1060 std::move(upsample_bilinear2d), in quant_fusion_pattern_and_replacements()
|
| /aosp_15_r20/external/executorch/backends/xnnpack/operators/ |
| H A D | op_static_resize_bilinear_2d.py | 26 target = "aten.upsample_bilinear2d.vec"
|
| /aosp_15_r20/external/executorch/backends/qualcomm/builders/ |
| H A D | op_upsample_bilinear2d.py | 19 target = ["aten.upsample_bilinear2d.default"]
|
| /aosp_15_r20/external/executorch/backends/qualcomm/_passes/ |
| H A D | convert_interpolate_with_upsample2d.py | 43 upsample_op = exir_ops.edge.aten.upsample_bilinear2d.default
|
| H A D | layout_transform.py | 41 exir_ops.edge.aten.upsample_bilinear2d.default,
|
| /aosp_15_r20/external/executorch/backends/xnnpack/_passes/ |
| H A D | convert_to_upsample_bilinear2d.py | 39 exir_ops.edge.aten.upsample_bilinear2d.vec,
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | decompositions.py | 3627 @register_decomposition(aten.upsample_bilinear2d.vec)
3631 @aten.upsample_bilinear2d.vec.py_impl(DispatchKey.CompositeImplicitAutograd)
3632 @aten.upsample_bilinear2d.vec.py_impl(DispatchKey.Autograd)
3653 [aten.upsample_bilinear2d.default, aten.upsample_bilinear2d.out]
3655 @aten.upsample_bilinear2d.default.py_impl(DispatchKey.Autograd)
3657 def upsample_bilinear2d( function
|
| /aosp_15_r20/external/executorch/backends/xnnpack/partition/ |
| H A D | configs.py | 43 exir_ops.edge.aten.upsample_bilinear2d.default,
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | coverage.yaml | 620 - aten::upsample_bilinear2d.vec
1002 aten::upsample_bilinear2d.vec: 8
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/utils/ |
| H A D | op_registry.cpp | 48 …"aten::upsample_bilinear2d(Tensor self, int[] output_size, bool align_corners, float? scales_h, fl… in nn_ops_first_input_preserving()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | UpSampleBilinear2d.cpp | 201 input_contig.scalar_type(), "upsample_bilinear2d", [&] { in upsample_bilinear2d_quantized_cpu()
|
| /aosp_15_r20/external/pytorch/test/export/ |
| H A D | testing.py | 95 aten.upsample_bilinear2d.vec,
|
| /aosp_15_r20/external/pytorch/test/edge/ |
| H A D | selected_operators.yaml | 428 aten::upsample_bilinear2d.vec_out:
|
| /aosp_15_r20/external/pytorch/functorch/op_analysis/ |
| H A D | annotated_ops | 478 upsample_bilinear2d, misc
|
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_dtype_analysis.py | 79 "upsample_bilinear2d",
|