Home
last modified time | relevance | path

Searched refs:is_built_with_cuda (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/
Dcudnn_rnn_test.py272 @unittest.skipUnless(test.is_built_with_cuda(),
325 @unittest.skipUnless(test.is_built_with_cuda(),
372 @unittest.skipUnless(test.is_built_with_cuda(),
415 @unittest.skipUnless(test.is_built_with_cuda(),
439 @unittest.skipUnless(test.is_built_with_cuda(),
672 @unittest.skipUnless(test.is_built_with_cuda(),
691 @unittest.skipUnless(test.is_built_with_cuda(),
696 @unittest.skipUnless(test.is_built_with_cuda(),
701 @unittest.skipUnless(test.is_built_with_cuda(),
706 @unittest.skipUnless(test.is_built_with_cuda(),
[all …]
Dcudnn_rnn_ops_test.py385 @unittest.skipUnless(test.is_built_with_cuda(),
409 @unittest.skipUnless(test.is_built_with_cuda(),
436 @unittest.skipUnless(test.is_built_with_cuda(),
468 @unittest.skipUnless(test.is_built_with_cuda(),
505 @unittest.skipUnless(test.is_built_with_cuda(),
767 @unittest.skipUnless(test.is_built_with_cuda(),
791 @unittest.skipUnless(test.is_built_with_cuda(),
818 @unittest.skipUnless(test.is_built_with_cuda(),
846 @unittest.skipUnless(test.is_built_with_cuda(),
878 @unittest.skipUnless(test.is_built_with_cuda(),
[all …]
/external/tensorflow/tensorflow/examples/adding_an_op/
Dcuda_op.py24 if tf.test.is_built_with_cuda():
Dcuda_op_test.py28 if tf.test.is_built_with_cuda():
/external/tensorflow/tensorflow/python/platform/
Dbuild_info_test.py28 self.assertEqual(build_info.is_cuda_build, test.is_built_with_cuda())
Dtest.py94 def is_built_with_cuda(): function
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.test.pbtxt32 name: "is_built_with_cuda"
/external/tensorflow/tensorflow/python/kernel_tests/
Dsoftmax_op_test.py129 @unittest.skipUnless(test.is_built_with_cuda(),
144 @unittest.skipUnless(test.is_built_with_cuda(),
Ddense_update_ops_test.py74 if test.is_built_with_cuda() and dtype in [np.float32, np.float64]:
Dsparse_xent_op_test.py90 if test.is_built_with_cuda() and test.is_gpu_available():
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.test.pbtxt48 name: "is_built_with_cuda"
/external/tensorflow/
DRELEASE.md1997 `tf.test.get_temp_dir` and `tf.test.is_built_with_cuda` for PEP-8