diff options
Diffstat (limited to 'ethosu/vela/tflite_mapping.py')
| -rw-r--r-- | ethosu/vela/tflite_mapping.py | 17 |
1 files changed, 7 insertions, 10 deletions
diff --git a/ethosu/vela/tflite_mapping.py b/ethosu/vela/tflite_mapping.py index e2b90765..e8b40bdb 100644 --- a/ethosu/vela/tflite_mapping.py +++ b/ethosu/vela/tflite_mapping.py @@ -20,19 +20,11 @@ # Contains a mapping from the various TensorFlow Lite enums and options structs, generated by the FlatBuffer code # generator, to Vela's internal format. -import numpy as np import struct -from .data_type import DataType - -from .tflite.TensorType import TensorType -from .tflite.BuiltinOperator import BuiltinOperator -from .tflite.BuiltinOptions import BuiltinOptions - - -from .tflite.Padding import Padding -from .tflite.ActivationFunctionType import ActivationFunctionType +import numpy as np +from .data_type import DataType from .tflite import Conv2DOptions from .tflite import DepthwiseConv2DOptions from .tflite import ConcatEmbeddingsOptions @@ -132,6 +124,11 @@ from .tflite import ScatterNdOptions from .tflite import SegmentSumOptions from .tflite import SelectV2Options from .tflite import WhileOptions +from .tflite.TensorType import TensorType +from .tflite.BuiltinOperator import BuiltinOperator +from .tflite.BuiltinOptions import BuiltinOptions +from .tflite.Padding import Padding +from .tflite.ActivationFunctionType import ActivationFunctionType def inverse_map(map): |