Function tvm::prim::cos