diff --git a/tfjs-backend-webgl/src/kernels/Conv2D_impl_test.ts b/tfjs-backend-webgl/src/kernels/Conv2D_impl_test.ts index 3631fca916e..63a94f87c4a 100644 --- a/tfjs-backend-webgl/src/kernels/Conv2D_impl_test.ts +++ b/tfjs-backend-webgl/src/kernels/Conv2D_impl_test.ts @@ -215,7 +215,6 @@ describeWithFlags('conv2dByMatMul', WEBGL_ENVS, () => { }); expect(result.shape).toEqual([1, 2, 2, 2]); - console.log(tf.backend().readSync(result.dataId)); expectArraysClose( tf.backend().readSync(result.dataId), [-11, -14, -17, -20, -110, -140, -170, -200]); @@ -235,8 +234,7 @@ describeWithFlags('conv2dByMatMul', WEBGL_ENVS, () => { const x = tf.tensor4d([1, 2, 3, 4, 5, 6, 7, 8], inShape); const w = tf.tensor4d( [-1, -1, -2, -2], [fSize, fSize, inputDepth, outputDepth]); - const alpha = - tf.tensor3d([0.001, 0.01, 0.1, 1, 10, 100, 1000, 10000], [2, 2, 2]); + const alpha = tf.tensor3d([1, 2, 3, 4, 5, 6, 7, 8], [2, 2, 2]); const $dataFormat = backend_util.convertConv2DDataFormat(dataFormat); const convInfo = backend_util.computeConv2DInfo( @@ -254,10 +252,9 @@ describeWithFlags('conv2dByMatMul', WEBGL_ENVS, () => { }); expect(result.shape).toEqual([1, 2, 2, 2]); - console.log(tf.backend().readSync(result.dataId)); expectArraysClose( tf.backend().readSync(result.dataId), - [-0.011, -0.14, -1.7, -20, -110, -1400, -17000, -200000]); + [-11, -28, -51, -80, -55, -84, -119, -160]); }); }); @@ -432,8 +429,7 @@ describeWithFlags('conv2dWithIm2Row', WEBGL_ENVS, () => { const w = tf.tensor4d( [-1, -1, -1, -1, 1, 1, 1, 1, -2, -2, -2, -2, 0.5, 0.5, 0.5, 0.5], [fSize, fSize, inputDepth, outputDepth]); - const alpha = - tf.tensor3d([0.001, 0.01, 0.1, 1, 10, 100, 1000, 10000], [2, 2, 2]); + const alpha = tf.tensor3d([1, 2, 3, 4, 5, 6, 7, 8], [2, 2, 2]); const $dataFormat = backend_util.convertConv2DDataFormat(dataFormat); const convInfo = backend_util.computeConv2DInfo( @@ -453,6 +449,6 @@ describeWithFlags('conv2dWithIm2Row', WEBGL_ENVS, () => { expect(result.shape).toEqual([1, 2, 2, 2]); expectArraysClose( tf.backend().readSync(result.dataId), - [-0.012, -0.32, 2, -12, -120, -3200, 2, -120000]); + [-12, -64, 2, -48, -60, -192, 2, -96]); }); });