Add getter functions in TF2 Python API

neo-ai / neo-ai-dlr

Neo-AI-DLR is a common runtime for machine learning models compiled by AWS SageMaker Neo, TVM, or TreeLite.

Apache License 2.0

492 stars 106 forks source link

Closed ghost closed 2 years ago

ghost commented 2 years ago

This PR adds missing TF2ModelImpl functions

get_input_dtype
get_input_dtypes
get_input_name
get_output_dtype
get_output_dtypes
get_output_name
has_metadata