Function tvm::prim::log2