iree-org / iree-org/iree

Squirrel across tflite tensor type in `iree.reflection`

Open
#3,978 1 comment 0 reactions 0 assignees View on GitHub
bindings/tflite
Dominant language
C++
Stars
3.9k
Forks
1k
Avg merge
4d 16h
Merged PRs (30d)
47

Description

We need the tflite canonical type for each I/O tensor. They're not just normal primitives but also `kTfLiteComplex64` and such. For now float32 is fine, but quantization #3972 will require the actual type.

Contributor guide

Open the contributing guide

Research direction

Start by locating the iree.reflection handling for TFLite I/O tensors and trace how their types are currently represented. Review the relationship to quantization issue #3972 and the TFLite types mentioned here. Done means each I/O tensor exposes its canonical type, including types such as kTfLiteComplex64 rather than assuming float32.

Written by the indexing model from the issue text.

Assessment

Tech stack
cpp, tensorflow
Domain
compilers, machine-learning
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.