diff --git a/tfjs-layers/src/layers/preprocessing/image_resizing_test.ts b/tfjs-layers/src/layers/preprocessing/image_resizing_test.ts index 5c490eb8eb6..ee632818d16 100644 --- a/tfjs-layers/src/layers/preprocessing/image_resizing_test.ts +++ b/tfjs-layers/src/layers/preprocessing/image_resizing_test.ts @@ -86,7 +86,7 @@ describeMathCPUAndGPU('Resizing Layer', () => { // create a resizing layer with same shape as input const height = 64; const width = 32; - const numChannels = 3; + const numChannels = 1; const rangeArr = [...Array(height * width).keys()]; const inputArr = []; while(rangeArr.length) {