tvm_tirx.tirx.script.masked_store

Contents

tvm_tirx.tirx.script.masked_store#

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

Store vector lanes selected by a predicate mask.

Parameters:

  • tensor (Var) – The tensor to update.

  • value (Expr) – The vector value to store.

  • 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 value.

  • loc (Location)

Returns:

call (Expr) – A void-typed tirx.masked_store call.

Contents