Function tvm::tirx::SubstituteWithDataTypeLegalization(PrimExpr, ffi::TypedFunction(const Var&)>)

Function tvm::tirx::SubstituteWithDataTypeLegalization(PrimExpr, ffi::TypedFunction<ffi::Optional<PrimExpr>(const Var&)>)#

Function Documentation#

PrimExpr tvm::tirx::SubstituteWithDataTypeLegalization(PrimExpr expr, ffi::TypedFunction<ffi::Optional<PrimExpr>(const Var&)> vmap)#

Substitute the var specified by vmap and legalize data types after substitution.

Substitution may change the data type of the expression.

Parameters:

  • expr – The source statement to be substituted

  • vmap – returns a new value if re-mapping is needed, otherwise returns nullptr.

Returns:

The result.