Files
apache--tvm/python
HoYi ccd81f220f [Relax][Frontend][TFLite] Fix dynamic FILL/SPLIT_V partial implementations (#19433)
This PR fixes partial TFLite frontend support for dynamic `FILL` and
`SPLIT_V`.

Key changes:
- Allow `FILL` to accept runtime `dims` tensors by converting them with
  `relax.op.tensor_to_shape` before calling `relax.op.full`.
- Allow `SPLIT_V` to accept runtime `size_splits` tensors by decomposing
the
  op into `cumsum` and `dynamic_strided_slice`.
- Use the TFLite output tuple arity as the source of truth for dynamic
`SPLIT_V`, instead of relying on `size_splits` static shape information.
- Add TFLite frontend regression tests covering dynamic `FILL` import
and
  dynamic `SPLIT_V` import/compile behavior.

This addresses the `FILL` and `SPLIT_V` items under the "Fix partial
implementations" section of #19412.

Validation:
```bash
python -m ruff check python/tvm/relax/frontend/tflite/tflite_frontend.py tests/python/relax/test_frontend_tflite.py
```

```bash
python -m pytest tests/python/relax/test_frontend_tflite.py \
-k "split_v_dynamic or fill_dynamic_dims" -v
```

Result:
- All checks passed
2026-04-28 14:45:25 -04:00
..