| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Upsample.cpp | 12 static Tensor upsample_nearest2d( in upsample_nearest2d() function
76 : VK_KERNEL(upsample_nearest2d), in upsample_nearest2d()
183 TORCH_SELECTIVE_NAME("aten::upsample_nearest2d"), in TORCH_LIBRARY_IMPL()
184 TORCH_FN(upsample_nearest2d)); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/executorch/backends/arm/test/ops/ |
| H A D | test_upsample_nearest2d.py | 93 .check(["torch.ops.aten.upsample_nearest2d.vec"])
96 .check_not(["torch.ops.aten.upsample_nearest2d.vec"])
118 .check(["torch.ops.aten.upsample_nearest2d.vec"])
121 .check_not(["torch.ops.aten.upsample_nearest2d.vec"])
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | UpSampleNearest2d.cpp | 16 #include <ATen/ops/upsample_nearest2d.h>
24 TORCH_META_FUNC(upsample_nearest2d) ( in TORCH_META_FUNC() argument
149 Tensor upsample_nearest2d( in upsample_nearest2d() function
156 return at::upsample_nearest2d(input, osize, scale_h, scale_w); in upsample_nearest2d()
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_caffe2.py | 19 "upsample_nearest2d",
205 def upsample_nearest2d( function
214 …return opset9.upsample_nearest2d(g, input, output_size, align_corners) # type: ignore[attr-define…
|
| H A D | symbolic_opset8.py | 74 "aten::upsample_nearest2d",
75 decorate=[symbolic_helper._apply_params("upsample_nearest2d", 4, "nearest")],
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/glsl/ |
| H A D | upsample_nearest2d.yaml | 7 upsample_nearest2d:
18 - NAME: upsample_nearest2d
|
| /aosp_15_r20/external/executorch/backends/arm/quantizer/quantization_annotation/ |
| H A D | upsample_nearest2d_annotator.py | 24 is_upsample = node.target == torch.ops.aten.upsample_nearest2d.vec
33 @register_annotator("upsample_nearest2d")
|
| /aosp_15_r20/external/executorch/backends/mediatek/ |
| H A D | partitioner.py | 77 torch.ops.aten.upsample_nearest2d.default,
78 torch.ops.aten.upsample_nearest2d.vec,
|
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/exporter/ |
| H A D | _decomp.py | 61 aten.upsample_nearest2d.default,
62 aten.upsample_nearest2d.vec,
|
| /aosp_15_r20/external/pytorch/test/cpp/tensorexpr/ |
| H A D | test_quantization.cpp | 325 %qu : QUInt8(1, 1, 6, 6) = aten::upsample_nearest2d(%q, %3, %4) in TEST_F()
333 auto qu = at::upsample_nearest2d(q, {6, 6}); in TEST_F()
359 %u : Float(1, 1, 4, 4) = aten::upsample_nearest2d(%x, %3, %4) in TEST_F()
365 auto y_expected = at::upsample_nearest2d(x, {4, 4}); in TEST_F()
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | Upsample.cpp | 93 std::string kernel_name("upsample_nearest2d"); in add_upsample_nearest2d_node()
120 VK_REGISTER_OP(aten.upsample_nearest2d.vec, upsample);
|
| /aosp_15_r20/external/executorch/kernels/aten/ |
| H A D | functions.yaml | 406 - op: upsample_nearest2d.out
408 - op: upsample_nearest2d.vec_out
|
| /aosp_15_r20/external/executorch/exir/backend/test/ |
| H A D | op_partitioner_demo.py | 182 torch.ops.aten.upsample_nearest2d.vec,
188 exir_ops.edge.aten.upsample_nearest2d.vec,
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | UpSampleNearest2d.cpp | 158 AT_DISPATCH_QINT_TYPES(input.scalar_type(), "upsample_nearest2d", [&] { in _upsample_nearest2d_quantized_cpu()
183 AT_DISPATCH_QINT_TYPES(input_contig.scalar_type(), "upsample_nearest2d", [&] { in _upsample_nearest2d_quantized_cpu()
|
| /aosp_15_r20/external/pytorch/test/edge/ |
| H A D | selected_operators.yaml | 434 aten::upsample_nearest2d.out:
440 aten::upsample_nearest2d.vec_out:
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesDecompositions.cpp | 345 m.impl("upsample_nearest2d.vec", native::upsample_nearest2d); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/operators/ |
| H A D | quantization.cpp | 705 "upsample_nearest2d", in computeUpsampleNearest2d()
751 "upsample_nearest2d", in computeUpsampleNearest2dExternalCall()
757 return BufHandle("upsample_nearest2d", outputShape, dtype); in computeUpsampleNearest2dExternalCall()
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | model_ops.yaml | 382 aten::upsample_nearest2d: 13
383 aten::upsample_nearest2d.vec: 52
|
| H A D | coverage.yaml | 623 - aten::upsample_nearest2d.vec
1005 aten::upsample_nearest2d.vec: 30
1075 aten::upsample_nearest2d: 7
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/metal/mpscnn/tests/ |
| H A D | MPSCNNTests.mm | 842 auto Y1 = at::upsample_nearest2d(
847 auto Y2 = at::upsample_nearest2d(
860 auto Y1 = at::upsample_nearest2d(
865 auto Y2 = at::upsample_nearest2d(
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | UpSampleNearest2d.cu | 283 // upsample_nearest2d meta call makes sure input/output tensor is not empty; in upsample_nearest2d_out_cuda_template()
406 // upsample_nearest2d meta call makes sure `nbatch != 0` in upsample_nearest2d_backward_out_cuda_template()
|
| /aosp_15_r20/external/pytorch/test/cpp/jit/ |
| H A D | test_flatbuffer.cpp | 77 return torch.upsample_nearest2d(input, [1, 1], float(scale), float(scale)) in TEST()
105 return torch.upsample_nearest2d(input, [1, 1], float(scale), float(scale)) in TEST()
210 return torch.upsample_nearest2d(input, [1, 1], float(scale), float(scale)) in TEST()
1223 return torch.upsample_nearest2d(input, [1, 1], float(scale), float(scale)) in TEST()
1321 return torch.upsample_nearest2d(input, [1, 1], float(scale), float(scale)) in TEST()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/ |
| H A D | quantization_patterns.h | 845 "aten::upsample_nearest2d", {"%output_size", "%scale_factors"}); in quant_fusion_pattern_and_replacements()
865 auto upsample_nearest2d = getInputTensorQParamOpFusionInfo( in quant_fusion_pattern_and_replacements() local
866 "aten::upsample_nearest2d", {"%output_size", "%scale_h", "%scale_w"}); in quant_fusion_pattern_and_replacements()
1057 std::move(upsample_nearest2d), in quant_fusion_pattern_and_replacements()
|
| /aosp_15_r20/external/executorch/backends/qualcomm/builders/ |
| H A D | op_upsample_nearest2d.py | 19 target = ["aten.upsample_nearest2d.default"]
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | decompositions.py | 2765 @register_decomposition(aten.upsample_nearest2d.vec)
2769 @aten.upsample_nearest2d.vec.py_impl(DispatchKey.CompositeImplicitAutograd)
2770 @aten.upsample_nearest2d.vec.py_impl(DispatchKey.Autograd)
2864 @register_decomposition([aten.upsample_nearest2d.default, aten.upsample_nearest2d.out])
2865 @aten.upsample_nearest2d.default.py_impl(DispatchKey.CompositeImplicitAutograd)
2866 @aten.upsample_nearest2d.default.py_impl(DispatchKey.Autograd)
2868 def upsample_nearest2d( function
|