Function tvm::prim::sqrt