aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/tflite_writer.py
diff options
context:
space:
mode:
authorDiego Russo <diego.russo@arm.com>2020-04-21 17:39:10 +0100
committerTim Hall <tim.hall@arm.com>2020-06-18 17:53:52 +0100
commite8a10454eba4c7392cb301fbfbe796e5bfb2b729 (patch)
tree8e54bffd46c55848a86040d3588b5365ef019d8f /ethosu/vela/tflite_writer.py
parent3ec04ac9e38d26193e0081a8e0fa3b8b667bb688 (diff)
downloadethos-u-vela-e8a10454eba4c7392cb301fbfbe796e5bfb2b729.tar.gz
Add reorder-python-import pre-commit hook
Also updated README.md Change-Id: I118309c61f4d00e8508d6b888c606995490fba39 Signed-off-by: Diego Russo <diego.russo@arm.com>
Diffstat (limited to 'ethosu/vela/tflite_writer.py')
-rw-r--r--ethosu/vela/tflite_writer.py33
1 files changed, 17 insertions, 16 deletions
diff --git a/ethosu/vela/tflite_writer.py b/ethosu/vela/tflite_writer.py
index 1f072424..99df849b 100644
--- a/ethosu/vela/tflite_writer.py
+++ b/ethosu/vela/tflite_writer.py
@@ -13,30 +13,31 @@
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
-
-
# Description:
# Functions used to write to a TensorFlow Lite format file. Supports adding in file identifiers.
-
-import numpy as np
import flatbuffers
-from flatbuffers.builder import UOffsetTFlags
-
-# ugh, the python flatbuffer interface is missing a method to add in file identifier. patching it in here:
import flatbuffers.number_types as N
+import numpy as np
from flatbuffers import encode
+from flatbuffers.builder import UOffsetTFlags
-from .tflite import Tensor
-from .tflite import QuantizationParameters
-from .tflite import Model
-from .tflite import SubGraph
-from .tflite import OperatorCode
-from .tflite import Operator
+from .nn_graph import PassPlacement
+from .tensor import MemArea
+from .tensor import TensorPurpose
from .tflite import Buffer
from .tflite import Metadata
-from .tflite_mapping import datatype_inv_map, builtin_operator_inv_map, custom_prefix, BuiltinOperator
-from .nn_graph import PassPlacement
-from .tensor import TensorPurpose, MemArea
+from .tflite import Model
+from .tflite import Operator
+from .tflite import OperatorCode
+from .tflite import QuantizationParameters
+from .tflite import SubGraph
+from .tflite import Tensor
+from .tflite_mapping import builtin_operator_inv_map
+from .tflite_mapping import BuiltinOperator
+from .tflite_mapping import custom_prefix
+from .tflite_mapping import datatype_inv_map
+
+# ugh, the python flatbuffer interface is missing a method to add in file identifier. patching it in here:
tflite_version = 3
tflite_file_identifier = "TFL" + str(tflite_version)