Skip to content

Commit 87816b9

Browse files
committed
Added TinyViT Slice layer test and generic slice parser quickfix
1 parent cbb2972 commit 87816b9

4 files changed

Lines changed: 1 addition & 1 deletion

File tree

Deeploy/Targets/Generic/Parsers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,7 @@ def parseNodeCtxt(self,
141141
ctxt.hoistConstant(axesTensor)
142142
node.inputs.append(axesTensor)
143143
if len(node.inputs) <= 4:
144-
values = np.ones((self.operatorRepresentation['dims']))
144+
values = np.ones((self.operatorRepresentation['dims']), dtype = np.int64)
145145
stepsTensor = gs.Constant(f'{node.name}_Steps_Tensor', values = values)
146146
ctxt.hoistConstant(stepsTensor)
147147
node.inputs.append(stepsTensor)
280 Bytes
Binary file not shown.
308 Bytes
Binary file not shown.
274 Bytes
Binary file not shown.

0 commit comments

Comments
 (0)