Function tvm::prim::atan