Function tvm::prim::rsqrt