Skip to content

Commit

Permalink
fix lint error (#18708)
Browse files Browse the repository at this point in the history
  • Loading branch information
guschmue authored Dec 5, 2023
1 parent 07aabcc commit 9aa7284
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions js/web/lib/wasm/jsep/webgpu/ops/pool.ts
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@ const getAdjustedPoolAttributesAndOutputShape = <AttributeType extends AveragePo
};

const getUniformAndPadInfo = <AttributeType extends AveragePoolAttributes|MaxPoolAttributes>(
xShape: readonly number[], outputShape: readonly number[],
outputShape: readonly number[],
attributes: AttributeType): [ProgramUniform[], UniformsArrayType, boolean, boolean, boolean] => {
const isChannelsLast = attributes.format === 'NHWC';
const outputSize = ShapeUtil.size(outputShape);
Expand Down Expand Up @@ -286,7 +286,7 @@ const createAveragePoolProgramInfo =
op2 += `value /= ${dataType}(i32(uniforms.kernelSize) - pad);`;
}
const [programUniforms, uniforms, hasPads, pwStartEnd, phStartEnd] =
getUniformAndPadInfo(input.dims, outputShape, adjustedAttributes);
getUniformAndPadInfo(outputShape, adjustedAttributes);
programUniforms.push(...createTensorShapeVariables(input.dims));
programUniforms.push(...createTensorShapeVariables(outputShape));
const inputDependencies: ProgramInputTensorInfoDependency[] = ['rank'];
Expand Down Expand Up @@ -362,7 +362,7 @@ const createMaxPoolProgramInfo =
const x = inputVariable('x', input.dataType, input.dims.length);
const inputDependencies: ProgramInputTensorInfoDependency[] = ['rank'];
const [programUniforms, uniforms, hasPads, pwStartEnd, phStartEnd] =
getUniformAndPadInfo(input.dims, outputShape, adjustedAttributes);
getUniformAndPadInfo(outputShape, adjustedAttributes);
programUniforms.push(...createTensorShapeVariables(input.dims));
programUniforms.push(...createTensorShapeVariables(outputShape));
return {
Expand Down

0 comments on commit 9aa7284

Please sign in to comment.