Home
last modified time | relevance | path

Searched full:upsample_bilinear2d (Results 1 – 25 of 55) sorted by relevance

123

/external/pytorch/torch/onnx/_internal/fx/
Ddecomposition_skip.py88 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]
/external/pytorch/aten/src/ATen/native/
DUpSampleBilinear2d.cpp18 #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()
Dts_native_functions.yaml139 - upsample_bilinear2d
/external/pytorch/aten/src/ATen/native/vulkan/ops/
DUpsample.cpp97 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()
/external/executorch/backends/mediatek/
Dpartitioner.py75 torch.ops.aten.upsample_bilinear2d.default,
76 torch.ops.aten.upsample_bilinear2d.vec,
/external/pytorch/torch/onnx/_internal/exporter/
D_decomp.py55 aten.upsample_bilinear2d.default,
56 aten.upsample_bilinear2d.vec,
/external/executorch/backends/qualcomm/builders/
Dop_upsample_bilinear2d.py19 target = ["aten.upsample_bilinear2d.default"]
/external/executorch/backends/xnnpack/operators/
Dop_static_resize_bilinear_2d.py26 target = "aten.upsample_bilinear2d.vec"
/external/executorch/backends/qualcomm/_passes/
Dconvert_interpolate_with_upsample2d.py43 upsample_op = exir_ops.edge.aten.upsample_bilinear2d.default
Dlayout_transform.py41 exir_ops.edge.aten.upsample_bilinear2d.default,
/external/executorch/backends/xnnpack/_passes/
Dconvert_to_upsample_bilinear2d.py39 exir_ops.edge.aten.upsample_bilinear2d.vec,
Dchannels_last_tagged_reshape_pass.py47 exir_ops.edge.aten.upsample_bilinear2d.vec,
/external/executorch/backends/xnnpack/partition/config/
Dgeneric_node_configs.py304 target_name = "upsample_bilinear2d.vec"
310 return torch.ops.aten.upsample_bilinear2d.vec
/external/pytorch/aten/src/ATen/functorch/
DBatchRulesDecompositions.cpp339 m.impl("upsample_bilinear2d.vec", native::upsample_bilinear2d); in TORCH_LIBRARY_IMPL()
/external/pytorch/torch/onnx/
Dsymbolic_opset8.py86 "aten::upsample_bilinear2d",
87 decorate=[symbolic_helper._apply_params("upsample_bilinear2d", 4, "linear")],
/external/executorch/backends/xnnpack/partition/
Dconfigs.py43 exir_ops.edge.aten.upsample_bilinear2d.default,
/external/pytorch/aten/src/ATen/native/quantized/cpu/
DUpSampleBilinear2d.cpp201 input_contig.scalar_type(), "upsample_bilinear2d", [&] { in upsample_bilinear2d_quantized_cpu()
/external/pytorch/torch/csrc/jit/passes/utils/
Dop_registry.cpp48 …"aten::upsample_bilinear2d(Tensor self, int[] output_size, bool align_corners, float? scales_h, fl… in nn_ops_first_input_preserving()
/external/pytorch/test/export/
Dtesting.py95 aten.upsample_bilinear2d.vec,
/external/pytorch/test/edge/
Dselected_operators.yaml428 aten::upsample_bilinear2d.vec_out:
/external/pytorch/functorch/op_analysis/
Dannotated_ops478 upsample_bilinear2d, misc
/external/pytorch/test/jit/
Dtest_dtype_analysis.py79 "upsample_bilinear2d",
/external/pytorch/aten/src/ATen/native/mps/operations/
DUpSample.mm19 #include <ATen/ops/upsample_bilinear2d.h>
/external/pytorch/torch/csrc/api/include/torch/nn/functional/
Dupsampling.h207 return torch::upsample_bilinear2d( in interpolate()
/external/pytorch/torch/csrc/jit/passes/quantization/
Dquantization_patterns.h855 "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()

123