Function tvm::tirx::masked_store_op

Function tvm::tirx::masked_store_op#

Function Documentation#

const Op &tvm::tirx::masked_store_op()#

Masked tensor store.

Arguments, in order:

  • args[0]: tensor, The tensor.

  • args[1]: value, The vector value to store.

  • args[2…n-2]: The tensor indices (at least one).

  • args[n-1]: The boolean lane mask, matching the lanes of value.