From e8a10454eba4c7392cb301fbfbe796e5bfb2b729 Mon Sep 17 00:00:00 2001 From: Diego Russo Date: Tue, 21 Apr 2020 17:39:10 +0100 Subject: Add reorder-python-import pre-commit hook Also updated README.md Change-Id: I118309c61f4d00e8508d6b888c606995490fba39 Signed-off-by: Diego Russo --- ethosu/vela/tflite_mapping.py | 171 +++++++++++++++++++++--------------------- 1 file changed, 84 insertions(+), 87 deletions(-) (limited to 'ethosu/vela/tflite_mapping.py') diff --git a/ethosu/vela/tflite_mapping.py b/ethosu/vela/tflite_mapping.py index e8b40bdb..4873ecc2 100644 --- a/ethosu/vela/tflite_mapping.py +++ b/ethosu/vela/tflite_mapping.py @@ -13,122 +13,119 @@ # 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: # TensorFlow Lite mapping functions used by both reader and writer. # Contains a mapping from the various TensorFlow Lite enums and options structs, generated by the FlatBuffer code # generator, to Vela's internal format. - import struct import numpy as np from .data_type import DataType -from .tflite import Conv2DOptions -from .tflite import DepthwiseConv2DOptions -from .tflite import ConcatEmbeddingsOptions -from .tflite import LSHProjectionOptions -from .tflite import Pool2DOptions -from .tflite import SVDFOptions -from .tflite import RNNOptions -from .tflite import FullyConnectedOptions -from .tflite import SoftmaxOptions -from .tflite import ConcatenationOptions +from .tflite import AbsOptions +from .tflite import AddNOptions from .tflite import AddOptions -from .tflite import L2NormOptions -from .tflite import LocalResponseNormalizationOptions -from .tflite import LSTMOptions -from .tflite import ResizeBilinearOptions -from .tflite import CallOptions -from .tflite import ReshapeOptions -from .tflite import SkipGramOptions -from .tflite import SpaceToDepthOptions -from .tflite import EmbeddingLookupSparseOptions -from .tflite import MulOptions -from .tflite import PadOptions -from .tflite import GatherOptions +from .tflite import ArgMaxOptions +from .tflite import ArgMinOptions from .tflite import BatchToSpaceNDOptions -from .tflite import SpaceToBatchNDOptions -from .tflite import TransposeOptions -from .tflite import ReducerOptions -from .tflite import SubOptions -from .tflite import DivOptions -from .tflite import SqueezeOptions -from .tflite import SequenceRNNOptions -from .tflite import StridedSliceOptions -from .tflite import ExpOptions -from .tflite import TopKV2Options -from .tflite import SplitOptions -from .tflite import LogSoftmaxOptions +from .tflite import BidirectionalSequenceLSTMOptions +from .tflite import BidirectionalSequenceRNNOptions +from .tflite import CallOptions from .tflite import CastOptions +from .tflite import ConcatEmbeddingsOptions +from .tflite import ConcatenationOptions +from .tflite import Conv2DOptions +from .tflite import CosOptions +from .tflite import DensifyOptions +from .tflite import DepthToSpaceOptions +from .tflite import DepthwiseConv2DOptions from .tflite import DequantizeOptions -from .tflite import MaximumMinimumOptions -from .tflite import ArgMaxOptions -from .tflite import LessOptions -from .tflite import NegOptions -from .tflite import PadV2Options -from .tflite import GreaterOptions -from .tflite import GreaterEqualOptions -from .tflite import LessEqualOptions -from .tflite import SelectOptions -from .tflite import SliceOptions -from .tflite import TransposeConvOptions -from .tflite import SparseToDenseOptions -from .tflite import TileOptions -from .tflite import ExpandDimsOptions +from .tflite import DivOptions +from .tflite import EmbeddingLookupSparseOptions from .tflite import EqualOptions -from .tflite import NotEqualOptions -from .tflite import ShapeOptions -from .tflite import PowOptions -from .tflite import ArgMinOptions +from .tflite import ExpandDimsOptions +from .tflite import ExpOptions from .tflite import FakeQuantOptions -from .tflite import PackOptions -from .tflite import LogicalOrOptions -from .tflite import OneHotOptions -from .tflite import LogicalAndOptions -from .tflite import LogicalNotOptions -from .tflite import UnpackOptions -from .tflite import FloorDivOptions -from .tflite import SquareOptions -from .tflite import ZerosLikeOptions from .tflite import FillOptions -from .tflite import BidirectionalSequenceLSTMOptions -from .tflite import BidirectionalSequenceRNNOptions -from .tflite import UnidirectionalSequenceLSTMOptions +from .tflite import FloorDivOptions from .tflite import FloorModOptions -from .tflite import RangeOptions -from .tflite import ResizeNearestNeighborOptions -from .tflite import LeakyReluOptions -from .tflite import SquaredDifferenceOptions -from .tflite import MirrorPadOptions -from .tflite import AbsOptions -from .tflite import SplitVOptions -from .tflite import UniqueOptions -from .tflite import ReverseV2Options -from .tflite import AddNOptions +from .tflite import FullyConnectedOptions from .tflite import GatherNdOptions -from .tflite import CosOptions -from .tflite import WhereOptions -from .tflite import RankOptions -from .tflite import ReverseSequenceOptions +from .tflite import GatherOptions +from .tflite import GreaterEqualOptions +from .tflite import GreaterOptions +from .tflite import IfOptions +from .tflite import L2NormOptions +from .tflite import LeakyReluOptions +from .tflite import LessEqualOptions +from .tflite import LessOptions +from .tflite import LocalResponseNormalizationOptions +from .tflite import LogicalAndOptions +from .tflite import LogicalNotOptions +from .tflite import LogicalOrOptions +from .tflite import LogSoftmaxOptions +from .tflite import LSHProjectionOptions +from .tflite import LSTMOptions from .tflite import MatrixDiagOptions -from .tflite import QuantizeOptions from .tflite import MatrixSetDiagOptions -from .tflite import DensifyOptions -from .tflite import DepthToSpaceOptions -from .tflite import IfOptions +from .tflite import MaximumMinimumOptions +from .tflite import MirrorPadOptions +from .tflite import MulOptions +from .tflite import NegOptions from .tflite import NonMaxSuppressionV4Options from .tflite import NonMaxSuppressionV5Options +from .tflite import NotEqualOptions +from .tflite import OneHotOptions +from .tflite import PackOptions +from .tflite import PadOptions +from .tflite import PadV2Options +from .tflite import Pool2DOptions +from .tflite import PowOptions +from .tflite import QuantizeOptions +from .tflite import RangeOptions +from .tflite import RankOptions +from .tflite import ReducerOptions +from .tflite import ReshapeOptions +from .tflite import ResizeBilinearOptions +from .tflite import ResizeNearestNeighborOptions +from .tflite import ReverseSequenceOptions +from .tflite import ReverseV2Options +from .tflite import RNNOptions from .tflite import ScatterNdOptions from .tflite import SegmentSumOptions +from .tflite import SelectOptions from .tflite import SelectV2Options +from .tflite import SequenceRNNOptions +from .tflite import ShapeOptions +from .tflite import SkipGramOptions +from .tflite import SliceOptions +from .tflite import SoftmaxOptions +from .tflite import SpaceToBatchNDOptions +from .tflite import SpaceToDepthOptions +from .tflite import SparseToDenseOptions +from .tflite import SplitOptions +from .tflite import SplitVOptions +from .tflite import SquaredDifferenceOptions +from .tflite import SquareOptions +from .tflite import SqueezeOptions +from .tflite import StridedSliceOptions +from .tflite import SubOptions +from .tflite import SVDFOptions +from .tflite import TileOptions +from .tflite import TopKV2Options +from .tflite import TransposeConvOptions +from .tflite import TransposeOptions +from .tflite import UnidirectionalSequenceLSTMOptions +from .tflite import UniqueOptions +from .tflite import UnpackOptions +from .tflite import WhereOptions from .tflite import WhileOptions -from .tflite.TensorType import TensorType +from .tflite import ZerosLikeOptions +from .tflite.ActivationFunctionType import ActivationFunctionType from .tflite.BuiltinOperator import BuiltinOperator from .tflite.BuiltinOptions import BuiltinOptions from .tflite.Padding import Padding -from .tflite.ActivationFunctionType import ActivationFunctionType +from .tflite.TensorType import TensorType def inverse_map(map): -- cgit v1.2.1