Method GetKernelSymbol

compiler/lib/KernelGen/BareMetal/BatchedMatmul.cpp:27–42

kernel gen

//! kernel gen
std::string BatchedMatrixMulKernel::GetKernelSymbol(TContext* context) const {
    std::stringstream ss;
    ss << "kernel_batched_matmul_";
    if (context->getAttrBool("transposeA")) {
        ss << "t";
    } else {
        ss << "n";
    }
    if (context->getAttrBool("transposeB")) {
        ss << "t";
    } else {
        ss << "n";
    }
    ss << "_" << context->getAttrOprand("operand:0").dtype;
    return ss.str();
}

namespace {
std::string emit_k(bool trans_a) {