Function tvm::prim::fma