HomeSort by relevance Sort by last modified time
    Searched refs:is_built_with_cuda (Results 1 - 8 of 8) sorted by null

  /external/tensorflow/tensorflow/examples/adding_an_op/
cuda_op.py 24 if tf.test.is_built_with_cuda():
cuda_op_test.py 28 if tf.test.is_built_with_cuda():
  /external/tensorflow/tensorflow/python/platform/
build_info_test.py 28 self.assertEqual(build_info.is_cuda_build, test.is_built_with_cuda())
test.py 28 @@is_built_with_cuda
105 @tf_export('test.is_built_with_cuda')
106 def is_built_with_cuda(): function
  /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/
cudnn_rnn_test.py 270 @unittest.skipUnless(test.is_built_with_cuda(),
323 @unittest.skipUnless(test.is_built_with_cuda(),
370 @unittest.skipUnless(test.is_built_with_cuda(),
413 @unittest.skipUnless(test.is_built_with_cuda(),
437 @unittest.skipUnless(test.is_built_with_cuda(),
664 @unittest.skipUnless(test.is_built_with_cuda(),
683 @unittest.skipUnless(test.is_built_with_cuda(),
688 @unittest.skipUnless(test.is_built_with_cuda(),
693 @unittest.skipUnless(test.is_built_with_cuda(),
698 @unittest.skipUnless(test.is_built_with_cuda(),
    [all...]
cudnn_rnn_ops_test.py 358 @unittest.skipUnless(test.is_built_with_cuda(),
395 @unittest.skipUnless(test.is_built_with_cuda(),
465 @unittest.skipUnless(test.is_built_with_cuda(),
615 @unittest.skipUnless(test.is_built_with_cuda(),
  /external/tensorflow/tensorflow/python/kernel_tests/
dense_update_ops_test.py 73 if test.is_built_with_cuda() and dtype in [np.float32, np.float64]:
sparse_xent_op_test.py 88 if test.is_built_with_cuda() and test.is_gpu_available():

Completed in 240 milliseconds