tvm_tirx.tirx.script.masked_load

Contents

tvm_tirx.tirx.script.masked_load#

tvm_tirx.tirx.script.masked_load(tensor, index, *args, ty=None, loc=UNKNOWN_LOC)#

Load vector lanes selected by a predicate mask.

Parameters:

  • ty (str) – The vector data type to load.

  • tensor (Var) – The tensor to load.

  • index (Expr) – The first tensor index.

  • args (Expr) – The tensor indices followed by a boolean lane mask. The mask must match the lane count and scalability of the loaded vector.

  • loc (Location)

Returns:

call (Expr) – A tirx.masked_load call with result type ty.

Contents