Method GetKernelSymbol

compiler/lib/CodeGen/Elemwise.cpp:43–49

}

std::string ElemwiseKernel::GetKernelSymbol(TContext* context) const {
    std::stringstream ss;
    ss << "mlir_auto_elementwise";
    ss << "_" << context->getAttrStr("mode");
    ss << "_" << context->getAttrOprand("operand:0").shape.size();
    return ss.str();
}

void ElemwiseKernel::CreateCompute(
        Block* entryBlock, mlir::OpBuilder& op_builder, mlir::MLIRContext* ctx,