diff options
author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-08-10 13:56:34 +0200 |
---|---|---|
committer | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-08-11 12:27:37 +0200 |
commit | d15866c06c88b9ec2e6313cc19f89ea65b528f8a (patch) | |
tree | 26dd354d0e49038ae0314445b60cece6e051701b /ethosu/vela/tosa | |
parent | ebb3b6fd78c54d6cf95b673aad3d868b798f91c7 (diff) | |
download | ethos-u-vela-d15866c06c88b9ec2e6313cc19f89ea65b528f8a.tar.gz |
MLBEDSW-4838 TOSA const data input changes
Adoptions related to changes for constant data
in TOSA.
Constant data not longer stored in .npy files, but
within the .tosa-file.
Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: Ia1148c2f8b783b3926a1ee0b9ad0a3aeff9d22f5
Diffstat (limited to 'ethosu/vela/tosa')
-rw-r--r-- | ethosu/vela/tosa/TosaTensor.py | 24 |
1 files changed, 20 insertions, 4 deletions
diff --git a/ethosu/vela/tosa/TosaTensor.py b/ethosu/vela/tosa/TosaTensor.py index 2a397db2..01ae7ece 100644 --- a/ethosu/vela/tosa/TosaTensor.py +++ b/ethosu/vela/tosa/TosaTensor.py @@ -55,16 +55,32 @@ class TosaTensor(object): return 0 # TosaTensor - def NpyFilename(self): + def Data(self, j): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10)) if o != 0: - return self._tab.String(o + self._tab.Pos) - return None + a = self._tab.Vector(o) + return self._tab.Get(flatbuffers.number_types.Uint8Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 1)) + return 0 + + # TosaTensor + def DataAsNumpy(self): + o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10)) + if o != 0: + return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Uint8Flags, o) + return 0 + + # TosaTensor + def DataLength(self): + o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10)) + if o != 0: + return self._tab.VectorLen(o) + return 0 def TosaTensorStart(builder): builder.StartObject(4) def TosaTensorAddName(builder, name): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(name), 0) def TosaTensorAddShape(builder, shape): builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(shape), 0) def TosaTensorStartShapeVector(builder, numElems): return builder.StartVector(4, numElems, 4) def TosaTensorAddType(builder, type): builder.PrependUint32Slot(2, type, 0) -def TosaTensorAddNpyFilename(builder, npyFilename): builder.PrependUOffsetTRelativeSlot(3, flatbuffers.number_types.UOffsetTFlags.py_type(npyFilename), 0) +def TosaTensorAddData(builder, data): builder.PrependUOffsetTRelativeSlot(3, flatbuffers.number_types.UOffsetTFlags.py_type(data), 0) +def TosaTensorStartDataVector(builder, numElems): return builder.StartVector(1, numElems, 1) def TosaTensorEnd(builder): return builder.EndObject() |