Method GetKernelSymbol

compiler/lib/KernelGen/BareMetal/Reduce.cpp:82–90

kernel gen

//! kernel gen
std::string ReduceKernel::GetKernelSymbol(TContext* context) const {
    std::stringstream ss;
    ss << "kernel_reduce";
    ss << "_" << context->getAttrStr("mode");
    ss << "_" << context->getAttrStr("data_type");
    ss << "_a" << context->getAttrInt("axis");
    ss << "_" << SymbolHelper::gen_io_str(context);
    return ss.str();
}

std::string ReduceKernel::GetKernelBody(TContext* context) const {
    auto mode = context->getAttrStr("mode");