Home
last modified time | relevance | path

Searched refs:cuda_only (Results 1 – 25 of 35) sorted by relevance

12

/external/tensorflow/tensorflow/python/grappler/
Dlayout_optimizer_test.py259 if test.is_gpu_available(cuda_only=True):
289 if test.is_gpu_available(cuda_only=True):
325 if test.is_gpu_available(cuda_only=True):
360 if test.is_gpu_available(cuda_only=True):
395 if test.is_gpu_available(cuda_only=True):
425 if test.is_gpu_available(cuda_only=True):
456 if test.is_gpu_available(cuda_only=True):
487 if test.is_gpu_available(cuda_only=True):
518 if test.is_gpu_available(cuda_only=True):
549 if test.is_gpu_available(cuda_only=True):
[all …]
Dconstant_folding_test.py54 if test.is_gpu_available(cuda_only=True):
/external/tensorflow/tensorflow/python/ops/
Dnn_fused_batchnorm_test.py345 if test.is_gpu_available(cuda_only=True):
355 if test.is_gpu_available(cuda_only=True):
364 if test.is_gpu_available(cuda_only=True):
372 if test.is_gpu_available(cuda_only=True):
383 if test.is_gpu_available(cuda_only=True):
394 if test.is_gpu_available(cuda_only=True):
405 if test.is_gpu_available(cuda_only=True):
413 if test.is_gpu_available(cuda_only=True):
421 if test.is_gpu_available(cuda_only=True):
432 if test.is_gpu_available(cuda_only=True):
[all …]
Dspectral_ops_test_util.py27 if test.is_gpu_available(cuda_only=True):
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_test.py157 if test.is_gpu_available(cuda_only=True):
329 if gpu_only and not test.is_gpu_available(cuda_only=True):
714 if use_gpu and not test.is_gpu_available(cuda_only=True):
1166 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():
1181 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():
1196 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():
1211 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():
1226 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():
1241 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():
1256 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():
[all …]
Dconv1d_transpose_test.py155 if test.is_gpu_available(cuda_only=True):
186 if test.is_gpu_available(cuda_only=True):
218 if test.is_gpu_available(cuda_only=True):
Dconv_ops_3d_test.py44 if test.is_gpu_available(cuda_only=True):
223 if test.is_gpu_available(cuda_only=True):
248 if test.is_gpu_available(cuda_only=True):
745 if test.is_gpu_available(cuda_only=True):
760 if test.is_gpu_available(cuda_only=True):
Dtranspose_op_test.py153 if not test.is_gpu_available(cuda_only=True):
177 if not test.is_gpu_available(cuda_only=True):
233 if not test.is_gpu_available(cuda_only=True):
255 if not test.is_gpu_available(cuda_only=True):
276 if not test.is_gpu_available(cuda_only=True):
Dconv2d_transpose_test.py180 if test.is_gpu_available(cuda_only=True):
215 if test.is_gpu_available(cuda_only=True):
251 if test.is_gpu_available(cuda_only=True):
Dsoftmax_op_test.py132 if test.is_gpu_available(cuda_only=True):
147 if test.is_gpu_available(cuda_only=True):
Dconv2d_backprop_filter_grad_test.py70 if test.is_gpu_available(cuda_only=True):
Dpooling_ops_test.py62 if not test.is_gpu_available(cuda_only=True):
70 cuda_only=True, min_cuda_compute_capability=(6, 1)):
908 if not test.is_gpu_available(cuda_only=True):
1923 if not test.is_gpu_available(cuda_only=True):
1934 if not test.is_gpu_available(cuda_only=True):
1947 if not test.is_gpu_available(cuda_only=True):
/external/tensorflow/tensorflow/python/keras/layers/
Dconvolutional_transpose_test.py57 if 'data_format' not in kwargs or test.is_gpu_available(cuda_only=True):
149 if 'data_format' not in kwargs or test.is_gpu_available(cuda_only=True):
200 if test.is_gpu_available(cuda_only=True):
Dconvolutional_test.py124 if 'data_format' not in kwargs or test.is_gpu_available(cuda_only=True):
191 if 'data_format' not in kwargs or test.is_gpu_available(cuda_only=True):
243 if test.is_gpu_available(cuda_only=True):
743 if 'data_format' not in kwargs or test.is_gpu_available(cuda_only=True):
Dseparable_convolutional_test.py127 if 'data_format' not in kwargs or test.is_gpu_available(cuda_only=True):
Dpooling_test.py131 if test.is_gpu_available(cuda_only=True):
/external/tensorflow/tensorflow/python/profiler/internal/
Drun_metadata_test.py116 if not test.is_gpu_available(cuda_only=True):
132 if not test.is_gpu_available(cuda_only=True):
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.test.pbtxt37 …argspec: "args=[\'cuda_only\', \'min_cuda_compute_capability\'], varargs=None, keywords=None, defa…
/external/tensorflow/tensorflow/python/client/
Dvirtual_gpu_test.py209 if not test.is_gpu_available(cuda_only=True):
238 if not test.is_gpu_available(cuda_only=True):
Dtimeline_test.py91 if not test.is_gpu_available(cuda_only=True):
/external/tensorflow/tensorflow/python/kernel_tests/signal/
Dfft_ops_test.py182 if test.is_gpu_available(cuda_only=True):
230 has_gpu = test.is_gpu_available(cuda_only=True)
370 if test.is_gpu_available(cuda_only=True):
404 if test.is_gpu_available(cuda_only=True):
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.test.pbtxt53 …argspec: "args=[\'cuda_only\', \'min_cuda_compute_capability\'], varargs=None, keywords=None, defa…
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_gpu_test.py70 if test.is_gpu_available(cuda_only=True):
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dlayers_test.py627 if test.is_gpu_available(cuda_only=True):
723 if test.is_gpu_available(cuda_only=True):
742 if test.is_gpu_available(cuda_only=True):
761 if test.is_gpu_available(cuda_only=True):
781 if test.is_gpu_available(cuda_only=True):
801 if test.is_gpu_available(cuda_only=True):
819 if test.is_gpu_available(cuda_only=True):
837 if test.is_gpu_available(cuda_only=True):
855 if test.is_gpu_available(cuda_only=True):
873 if test.is_gpu_available(cuda_only=True):
[all …]
/external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/
Dbeam_search_ops_test.py77 if not test.is_gpu_available(cuda_only=True):

12