doc-ai / tensorio-ios

Tensor/IO for iOS, with support for on-device inference and training with TensorFlow and TensorFlow Lite
Other
31 stars 10 forks source link

Share i/o descriptions with model bundle #136

Closed phildow closed 5 years ago

phildow commented 5 years ago

Finding that we often need access to the inputs and outputs descriptions without the model itself. Move those descriptions to the model bundle, parse there, and reference them from the model itself.

I'm thinking we use a TensorIOInputsOutputs class that encompasses the inputs and outputs and places them on an io property on the model bundle and on the model, so you have:

bundle.io.inputs
bundle.io.outputs

And

model.io.inputs
model.io.outputs