diff options
| author | Diego Russo <diego.russo@arm.com> | 2020-04-14 18:41:58 +0100 |
|---|---|---|
| committer | Tim Hall <tim.hall@arm.com> | 2020-06-18 17:53:52 +0100 |
| commit | ea6111a36e55501bbfb9ea022aaf8305b4d80183 (patch) | |
| tree | f6787f7d07ea56a502616a8d6f36bc9c7e6e4efe /ethosu/vela/tflite_mapping.py | |
| parent | 2213e90570af328418d4f4a0d54269ed21dc40bc (diff) | |
| download | ethos-u-vela-ea6111a36e55501bbfb9ea022aaf8305b4d80183.tar.gz | |
Add pre-commit support for sanity checks
Use pre-commit framework [1] to run black and flake8 before the commit.
black and flake8 are managed by the pre-commit framework and they can be
run manually by the user using `pre-commit run` command.
Fix the code base with the help of black and flake8.
Fix import statements according to PEP8 guidelines [1]
Both tools have the following settings (specified in the pre-commit
configuration file):
* line length: 120 characters
* directory to exclude: ethosu/vela/tflite/ and ethosu/vela/ethos_u55_regs
Updated README.md on how to install pre-commit and how to run sanity checks.
Pipenv files have been updated including new dependencies for pre-commit.
[1]: https://www.python.org/dev/peps/pep-0008/#imports
[2]: https://github.com/pre-commit/pre-commit
Change-Id: I304d9fffdf019d390ffa396a529c8a7c2437f63d
Signed-off-by: Diego Russo <diego.russo@arm.com>
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): |