Method IsAvailable

compiler/lib/KernelGen/BareMetal/Argmax.cpp:34–37

namespace

}
}  // namespace
bool ArgmaxKernel::IsAvailable(TContext* context) const {
    bool ok_dtype = context->getAttrOprand("operand:0").dtype == "f32";
    return ok_dtype;
}
//! kernel gen
std::string ArgmaxKernel::GetKernelSymbol(TContext* context) const {
    std::stringstream ss;