Skip to content

Commit 08e92ed

Browse files
authored
reduce size of matrix in test (tensorflow#1426)
* reduces the size of a matrix in a test to speed unit tests
1 parent aa71e5a commit 08e92ed

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/ops/linalg_ops_test.ts

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -85,12 +85,12 @@ describeWithFlags('gramSchmidt-tiny', ALL_ENVS, () => {
8585
// For operations on non-trivial matrix sizes, we skip the CPU-only ENV and use
8686
// only WebGL ENVs.
8787
describeWithFlags('gramSchmidt-non-tiny', WEBGL_ENVS, () => {
88-
it('32x512', () => {
88+
it('16x128', () => {
8989
// Part of this test's point is that operation on a matrix of this size
9090
// can complete in the timeout limit of the unit test.
91-
const xs = tf.randomUniform([32, 512]) as Tensor2D;
91+
const xs = tf.randomUniform([16, 128]) as Tensor2D;
9292
const y = tf.linalg.gramSchmidt(xs) as Tensor2D;
93-
expectArraysClose(y.matMul(y.transpose()), tf.eye(32));
93+
expectArraysClose(y.matMul(y.transpose()), tf.eye(16));
9494
});
9595
});
9696

0 commit comments

Comments
 (0)