Home
last modified time | relevance | path

Searched refs:upsample_nearest2d_vec (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/metal/ops/
H A DMetalUpsamplingNearest.mm16 static Tensor upsample_nearest2d_vec(
68 m.impl(TORCH_SELECTIVE_NAME("aten::upsample_nearest2d.vec"), TORCH_FN(upsample_nearest2d_vec));
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/
H A Dquantization_patterns.h844 auto upsample_nearest2d_vec = getInputTensorQParamOpFusionInfo( in quant_fusion_pattern_and_replacements() local
1063 std::move(upsample_nearest2d_vec), in quant_fusion_pattern_and_replacements()