Add Transformer Clear method

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 apivovarov closed 2 years ago

apivovarov commented 2 years ago

Changes:

Add Transformer::Clear method to clear memory after InitNDArray and MapToNDArray cycle.
Remove "read-only" const post-fix from InitNDArray methods because they might change private members in their classes.
Add override tag to MapToNDArray method in child classes