mirror of https://github.com/llvm/torch-mlir
52be4bdc18
This commit fixes the bugs for the `onnx.OneHot` operator by: 1) Converting negative indices to non-negative indices 2) Handling both `int` and `float` types for `off` and `on` values 3) Using the correct result type It also includes a new unit test. |
||
---|---|---|
.. | ||
ops | ||
simple_ops_a_to_f.mlir | ||
simple_ops_g_to_p.mlir | ||
simple_ops_q_to_z.mlir | ||
unsupported_fb_opt_ops.mlir |