Method main

tests/python/contrib/test_example_npu.py:91–95

(x: R.Tensor((2, 8), "float32"))

    @R.function
    def main(x: R.Tensor((2, 8), "float32")) -> R.Tensor((2, 8), "float32"):
        with R.dataflow():
            z = relax.op.nn.softmax(x)
            R.output(z)
        return z


# Check if the example NPU runtime is available