diff options
author | Diego Russo <diego.russo@arm.com> | 2020-04-21 17:39:10 +0100 |
---|---|---|
committer | Tim Hall <tim.hall@arm.com> | 2020-06-18 17:53:52 +0100 |
commit | e8a10454eba4c7392cb301fbfbe796e5bfb2b729 (patch) | |
tree | 8e54bffd46c55848a86040d3588b5365ef019d8f /ethosu/vela/tflite_mapping.py | |
parent | 3ec04ac9e38d26193e0081a8e0fa3b8b667bb688 (diff) | |
download | ethos-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_mapping.py')
-rw-r--r-- | ethosu/vela/tflite_mapping.py | 171 |
1 files changed, 84 insertions, 87 deletions
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): |