Skip to content

Commit 4a70492

Browse files
BruceDaianssiko
authored andcommitted
Bugfix: Fix typos in dequantizeLinear's decomposition code
1 parent 20ed414 commit 4a70492

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

index.bs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3999,9 +3999,9 @@ partial dictionary MLOpSupportLimits {
39993999
// - inputShape = [2,3,4,5,6] with axis = 2 yields shape [6,4,30].
40004000
// - inputShape = [4] with axis = 0 yields shape [1,4,1].
40014001
function getFlattenedShapeAroundAxis(inputShape, axis) {
4002-
axis = Math.max(Math.min(axis, input.shape.length - 1), 0);
4002+
axis = Math.max(Math.min(axis, inputShape.length - 1), 0);
40034003
const shapeBefore = inputShape.slice(0, axis);
4004-
const shapeAfter = inputShape.slice(axis + 1, input.shape.length);
4004+
const shapeAfter = inputShape.slice(axis + 1, inputShape.length);
40054005
const countBefore = shapeBefore.reduce((a, b) => a * b, 1);
40064006
const countAfter = shapeAfter.reduce((a, b) => a * b, 1);
40074007
return [countBefore, inputShape[axis], countAfter];

0 commit comments

Comments
 (0)