tvm_tirx.ir.prim.indexmod

Contents

tvm_tirx.ir.prim.indexmod#

tvm_tirx.ir.prim.indexmod(a, b, loc=UNKNOWN_LOC)#

Compute the remainder of indexdiv. a and b are non-negative.

Parameters:

  • a (Expr) – The left hand operand, known to be non-negative.

  • b (Expr) – The right hand operand, known to be non-negative.

  • loc (Location, optional) – The location of this operator in the source.

Returns:

res (Expr) – The result expression.

Note

Use this function to split non-negative indices. This function may take advantage of operands’ non-negativeness.

Contents