For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).
Mojo struct
ArgNonZero
struct ArgNonZero
Registers the mo.arg_nonzero graph op with the graph compiler.
Implemented traits
Methods
execute
static def execute(output_buffer: ManagedTensorSlice[IOSpec[_, _].Output, static_spec=output_buffer.static_spec], input_buffer: ManagedTensorSlice[IOSpec[_, _].Input, static_spec=input_buffer.static_spec])
Executes the mo.arg_nonzero graph op.
Args:
- output_buffer (
ManagedTensorSlice[IOSpec[_, _].Output, static_spec=output_buffer.static_spec]): Output tensor receiving the nonzero indices. - input_buffer (
ManagedTensorSlice[IOSpec[_, _].Input, static_spec=input_buffer.static_spec]): Input tensor whose nonzero indices are found.