Searched refs:run_without_tensor_float_32 (Results 1 – 13 of 13) sorted by relevance
62 @test_util.run_without_tensor_float_32(104 @test_util.run_without_tensor_float_32(
198 @test_util.run_without_tensor_float_32("Tests Qr gradient, which calls matmul"
219 @test_util.run_without_tensor_float_32(232 @test_util.run_without_tensor_float_32(539 @test_util.run_without_tensor_float_32(550 @test_util.run_without_tensor_float_32(
91 @test_util.run_without_tensor_float_32('TF32 capable devices fail the test'178 @test_util.run_without_tensor_float_32('TF32 capable devices fail the test'
129 @test_util.run_without_tensor_float_32("Tests batch matmul")141 @test_util.run_without_tensor_float_32("Tests batch matmul")
164 @test_util.run_without_tensor_float_32("Tests tensordot, which calls matmul")196 @test_util.run_without_tensor_float_32("Tests tensordot, which calls matmul")
103 @test_util.run_without_tensor_float_32("Tests matmul")149 @test_util.run_without_tensor_float_32("Tests matmul")
977 def run_without_tensor_float_32(description): # pylint: disable=unused-argument function1012 return for_all_test_methods(run_without_tensor_float_32, description)
279 @test_util.run_without_tensor_float_32(767 @test_util.run_without_tensor_float_32(
199 @test_util.run_without_tensor_float_32(
267 @test_util.run_without_tensor_float_32(
2269 def run_without_tensor_float_32(description): # pylint: disable=unused-argument function2304 return for_all_test_methods(run_without_tensor_float_32, description)
115 @test_util.run_without_tensor_float_32(147 @test_util.run_without_tensor_float_32(