Example

invalid_argmax_device.kina

Expected rejection

This program is intentionally invalid. The compiler should reject its type or shape.

Source

tensor<F32, [2, 3], @device(0)>.reduce(op=argmax, axis=1)