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

Linter errors

Browse Source
This commit is contained in:
Eugene Kuznetsov 2020-01-15 13:15:04 -08:00
parent 543db6fc67
commit 0f0aa37512
6 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/python/kernel_tests/batch_matmul_op_test.py
View File

@ -262,7 +262,8 @@ class BatchMatMulBenchmark(test.Benchmark):
if __name__ == "__main__":
dtypes_to_test = [np.float16, np.float32, np.float64, np.int32, np.complex64, np.complex128]
dtypes_to_test = [np.float16, np.float32, np.float64, np.int32,
np.complex64, np.complex128]
for dtype_ in dtypes_to_test:
for adjoint_a_ in False, True:
for adjoint_b_ in False, True:

3
tensorflow/python/kernel_tests/eig_op_test.py
View File

@ -183,7 +183,8 @@ def _GetEigTest(dtype_, shape_, compute_v_):
if __name__ == "__main__":
dtypes_to_test = [dtypes_lib.float32, dtypes_lib.float64, dtypes_lib.complex64, dtypes_lib.complex128]
dtypes_to_test = [dtypes_lib.float32, dtypes_lib.float64,
dtypes_lib.complex64, dtypes_lib.complex128]
for compute_v in True, False:
for dtype in dtypes_to_test:
for size in 1, 2, 5, 10:

3
tensorflow/python/kernel_tests/linalg/linear_operator_adjoint_test.py
View File

@ -199,7 +199,8 @@ class LinearOperatorAdjointTest(
def test_solve_adjoint_complex_operator(self):
if test.is_built_with_rocm():
self.skipTest("ROCm does not support BLAS solve operations for complex types")
self.skipTest("ROCm does not support BLAS solve operations"
" for complex types")
matrix1 = self.evaluate(linear_operator_test_util.random_tril_matrix(
[4, 4], dtype=dtypes.complex128, force_well_conditioned=True) +
1j * linear_operator_test_util.random_tril_matrix(

3
tensorflow/python/kernel_tests/matmul_op_test.py
View File

@ -226,7 +226,8 @@ class MatMulInfixOperatorTest(test_lib.TestCase):
if __name__ == "__main__":
sizes = [1, 3, 5]
trans_options = [[False, False], [True, False], [False, True]]
dtypes_to_test = [np.int32, np.int64, np.float16, np.float32, np.float64, np.complex64, np.complex128]
dtypes_to_test = [np.int32, np.int64, np.float16, np.float32, np.float64,
np.complex64, np.complex128]
# TF2 does not support placeholders under eager so we skip it
for use_static_shape in set([True, tf2.enabled()]):
for dtype in dtypes_to_test:

3
tensorflow/python/kernel_tests/self_adjoint_eig_op_test.py
View File

@ -240,7 +240,8 @@ def _GetSelfAdjointEigGradTest(dtype_, shape_, compute_v_):
if __name__ == "__main__":
dtypes_to_test = [dtypes_lib.float32, dtypes_lib.float64, dtypes_lib.complex64, dtypes_lib.complex128]
dtypes_to_test = [dtypes_lib.float32, dtypes_lib.float64,
dtypes_lib.complex64, dtypes_lib.complex128]
for compute_v in True, False:
for dtype in dtypes_to_test:
for size in 1, 2, 5, 10:

3
tensorflow/python/kernel_tests/tensordot_op_test.py
View File

@ -221,7 +221,8 @@ def _get_tensordot_tests(dtype_, rank_a_, rank_b_, num_dims_, dynamic_shape_):
if __name__ == "__main__":
dtypes_to_test = [np.float16, np.float32, np.float64, np.complex64, np.complex128]
dtypes_to_test = [np.float16, np.float32, np.float64,
np.complex64, np.complex128]
for dtype in dtypes_to_test:
for rank_a in 1, 2, 4, 5:
for rank_b in 1, 2, 4, 5: