Searched refs:is_built_with_cuda (Results 1 – 8 of 8) sorted by relevance
93 def is_built_with_cuda(): function170 return is_built_with_cuda() or is_built_with_rocm()
29 test.is_built_with_cuda())
34 self.assertEqual(build_info["is_cuda_build"], test.is_built_with_cuda())
20 if tf.test.is_built_with_cuda():
24 if tf.test.is_built_with_cuda():
36 name: "is_built_with_cuda"
52 name: "is_built_with_cuda"
10038 `tf.test.get_temp_dir` and `tf.test.is_built_with_cuda` for PEP-8