tvm_tirx.ir.prim.round

Contents

tvm_tirx.ir.prim.round#

tvm_tirx.ir.prim.round(x, loc=UNKNOWN_LOC)#

Round elements of the array to the nearest integer.

Parameters:
  • x (Expr) – Input argument.

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

Returns:

y (Expr) – The result.