Skip to content

Commit cae8406

Browse files
committed
Format fix
1 parent 4d6f129 commit cae8406

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

Deeploy/Targets/PULPOpen/TileConstraints/ConvTileConstraint.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -251,7 +251,7 @@ def addGeometricalConstraint(tilerModel: TilerModel, parseDict: Dict, ctxt: Netw
251251
# Get to-be-tiled tensor buffers
252252
inputBufferName = parseDict['data_in']
253253
outputBufferName = parseDict['data_out']
254-
254+
255255
weightBufferName = parseDict['weight']
256256
biasBufferName = parseDict['bias']
257257

@@ -311,8 +311,10 @@ def addGeometricalConstraint(tilerModel: TilerModel, parseDict: Dict, ctxt: Netw
311311
effectiveHeight = inputHeightVar + ((pads[0] + pads[2]) * (inputHeightVar == inputBuffer.shape[1]))
312312
effectiveWidth = inputWidthVar + ((pads[1] + pads[3]) * (inputWidthVar == inputBuffer.shape[2]))
313313

314-
tilerModel.addConstraint((outputHeightVar == (effectiveHeight - dilations[0] * (weightHeightVar - 1) - 1) // strides[0] + 1))
315-
tilerModel.addConstraint((outputWidthVar == (effectiveWidth - dilations[1] * (weightWidthVar - 1) - 1) // strides[1] + 1))
314+
tilerModel.addConstraint(
315+
(outputHeightVar == (effectiveHeight - dilations[0] * (weightHeightVar - 1) - 1) // strides[0] + 1))
316+
tilerModel.addConstraint(
317+
(outputWidthVar == (effectiveWidth - dilations[1] * (weightWidthVar - 1) - 1) // strides[1] + 1))
316318

317319
# Add constraint for input channel size match
318320
# (Depends on weight output channel and conv grouping)

0 commit comments

Comments
 (0)