Method GetKernelSymbol

compiler/lib/KernelGen/BareMetal/IndexingOneHot.cpp:19–24

kernel gen

}
//! kernel gen
std::string IndexingOneHotKernel::GetKernelSymbol(TContext* context) const {
    std::stringstream ss;
    ss << "kernel_indexingonehot_" << context->getAttrInt("axis") << "_"
       << context->getAttrOprand("operand:0").dtype;
    return ss.str();
}

std::string IndexingOneHotKernel::GetKernelBody(TContext* context) const {
    std::stringstream axis_init_ss;