Function tvm::backend::metal::cooperative_tensor_fill_op

Contents

Function tvm::backend::metal::cooperative_tensor_fill_op#

Defined in File cooperative_tensor.h

Function Documentation#

const Op &tvm::backend::metal::cooperative_tensor_fill_op()#

Fill a cooperative_tensor with a given value.

Arguments, in order:

args[0]: d, The D operand.
args[1]: index, The index.
args[2]: value, The value to use.
args[3]: rows, The number of rows.
args[4]: cols, The number of columns.