experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Bug fix in tests for tf.linalg.LinearOperatorBlockDiag.

Browse Source 
PiperOrigin-RevId: 288054665
Change-Id: If06da1a443eb2fad0206809490a442cd5fe62b99
This commit is contained in:
Emily Fertig 2020-01-03 14:58:01 -08:00 committed by TensorFlower Gardener
parent 90fe9cbcb1
commit 55e68264ba
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/python/kernel_tests/linalg/linear_operator_block_diag_test.py
View File

@ -70,9 +70,9 @@ class SquareLinearOperatorBlockDiagTest(
self._rtol[dtypes.float32] = 1e-4
self._rtol[dtypes.complex64] = 1e-4
@property
def operator_shape_infos(self):
shape_info = linear_operator_test_util.OperatorShapeInfo
@staticmethod
def operator_shapes_infos():
shape_info = linear_operator_test_util.OperatorShapesInfo
return [
shape_info((0, 0)),
shape_info((1, 1)),