Skip to content

Commit fffe8f3

Browse files
committed
Fix reshape bindings (which are used for squeeze/unsqeeze too) to type map axes to int64_t as per onnx
1 parent 01e7788 commit fffe8f3

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

Deeploy/Targets/PULPOpen/Bindings.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,8 @@
99
from Deeploy.CommonExtensions.CodeTransformationPasses.Closure import ClosureGeneration, MemoryAwareClosureGeneration
1010
from Deeploy.CommonExtensions.CodeTransformationPasses.MemoryAllocation import ArgumentStructGeneration, \
1111
MemoryManagementGeneration, MemoryPassthroughGeneration
12-
from Deeploy.CommonExtensions.DataTypes import IntegerDataTypes, SignedIntegerDataTypes, float32_t, int8_t, int32_t, \
13-
uint8_t
12+
from Deeploy.CommonExtensions.DataTypes import FloatDataTypes, IntegerDataTypes, SignedIntegerDataTypes, float32_t, \
13+
int8_t, int32_t, int64_t, uint8_t
1414
from Deeploy.DeeployTypes import CodeTransformation, NodeBinding, NodeTemplate
1515
from Deeploy.FutureExtension.Bindings.AutoFutureBinding import AutoFutureBinding
1616
from Deeploy.FutureExtension.CodeTransformationPasses.FutureCodeTransformation import FutureGeneration
@@ -153,11 +153,8 @@
153153
]
154154

155155
PULPReshapeBindings = [
156-
NodeBinding(ReshapeChecker([PointerClass(type), PointerClass(int32_t)], [PointerClass(type)]),
157-
ReshapeTemplate.referenceTemplate, SkipTransformer) for type in IntegerDataTypes
158-
] + [
159-
NodeBinding(ReshapeChecker([PointerClass(float32_t), PointerClass(type)], [PointerClass(float32_t)]),
160-
ReshapeTemplate.referenceTemplate, SkipTransformer) for type in IntegerDataTypes
156+
NodeBinding(ReshapeChecker([PointerClass(type), PointerClass(int64_t)], [PointerClass(type)]),
157+
ReshapeTemplate.referenceTemplate, SkipTransformer) for type in IntegerDataTypes + FloatDataTypes
161158
]
162159

163160
PULPRQAddBindings = [

0 commit comments

Comments
 (0)