Method GetKernelBody

compiler/lib/KernelGen/Arm/ArmCommon/Resize.cpp:32–82

}

std::string ResizeKernel::GetKernelBody(TContext* context) const {
    std::stringstream ss;
    auto src_dtype = context->getAttrOprand("operand:0").dtype;
    auto fmt = context->getAttrStr("format");
    auto specifier = Utils::cvt_dtype_specifier(src_dtype);
    auto imode = context->getAttrStr("imode");
    ss << R"(
        #include <arm_neon.h>
        #include <math.h>
        #include <stdalign.h>
    )";
    auto coord_str = ResizeHelper::GenCoordHelper(imode, specifier);
    auto gen_layout_dims = ResizeHelper::GenLayoutDims(fmt);
    auto get_offset = ResizeHelper::GenGetOffset(fmt);
    ss << StringTemplate::StringTemplateArgs()
                    .add("coord_helper_str", coord_str)
                    .add("get_offset", get_offset)
                    .render(R"(
        static inline float output_converter(float x){
            return x;
        }
        ${coord_helper_str}
        ${get_offset}
        #define rep(i, n) for (int i = 0; i < (n); ++i)
    )");
    ss << GenCommonRet() << " " << GetKernelSignature(context);
    std::string body_temp = R"({
        const Tensor* src_tensor = inputs[0];
        const Tensor* dst_tensor = outputs[0];
        ${specifier}* sptr = (${specifier}*)(src_tensor->ptr);
        ${specifier}* dptr = (${specifier}*)(dst_tensor->ptr);
        TINYNN_ASSERT(sptr);
        TINYNN_ASSERT(dptr);
        
        const Layout src_layout = src_tensor->layout;
        const Layout dst_layout = dst_tensor->layout;
        ${gen_layout_dims}
        float scale_h = (float)(OH) / IH;
        float scale_w = (float)(OW) / IW;

        ${normal_impl}
        return TinyNN_SUCCESS;
    })";
    auto normal_impl = ResizeHelper::GenNormImpl(fmt);
    ss << StringTemplate::StringTemplateArgs()
                    .add("specifier", specifier)
                    .add("normal_impl", normal_impl)
                    .add("gen_layout_dims", gen_layout_dims)
                    .render(body_temp);
    return ss.str();
}

bool ResizeKernel::IsCVAvailable(TContext* context) const {
    auto src_dtype = context->getAttrOprand("operand:0").dtype;