This patch adds initial support for the tflite operator split. However I am not yet sure how to handle the axis parameter for the split operator and support it in the test infrastructure. Putting this up for an initial review and comment.
The split operator in tflite according to https://www.tensorflow.org/lite/guide/ops_compatibility appears to take num_or_size_split as a 0D tensor. I also note that tflite.split is one of the few operators that returns multiple outputs and thus the helper routines in the tests needed some massaging to make this work. @apivarov , could you please review this ? Thanks, Ramana You can view, comment on, or merge this pull request online at: https://github.com/dmlc/tvm/pull/3520 -- Commit Summary -- * [RFC] Initial support for Tflite operator SPLIT -- File Changes -- M python/tvm/relay/frontend/tflite.py (28) M tests/python/frontend/tflite/test_forward.py (31) -- Patch Links -- https://github.com/dmlc/tvm/pull/3520.patch https://github.com/dmlc/tvm/pull/3520.diff -- You are receiving this because you are subscribed to this thread. Reply to this email directly or view it on GitHub: https://github.com/dmlc/tvm/pull/3520
