Home
last modified time | relevance | path

Searched refs:should_use_gpu (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/keras/tests/
Dtracking_util_with_v1_optimizers_test.py392 with testing_utils.device(should_use_gpu=True):
425 with testing_utils.device(should_use_gpu=True):
497 with testing_utils.device(should_use_gpu=True):
524 with testing_utils.device(should_use_gpu=True):
546 with testing_utils.device(should_use_gpu=True):
629 with testing_utils.device(should_use_gpu=True):
Dtracking_util_test.py348 with testing_utils.device(should_use_gpu=True):
401 with testing_utils.device(should_use_gpu=True):
609 with testing_utils.device(should_use_gpu=True):
645 with testing_utils.device(should_use_gpu=True):
669 with testing_utils.device(should_use_gpu=True):
823 with testing_utils.device(should_use_gpu=True):
/external/tensorflow/tensorflow/python/keras/layers/
Dgru_v2_test.py190 with testing_utils.device(should_use_gpu=True):
253 with testing_utils.device(should_use_gpu=False):
260 with testing_utils.device(should_use_gpu=True):
270 with testing_utils.device(should_use_gpu=True):
587 with testing_utils.device(should_use_gpu=True):
594 with testing_utils.device(should_use_gpu=True):
Drecurrent_v2_test.py55 with testing_utils.device(should_use_gpu=False):
Dlstm_v2_test.py362 with testing_utils.device(should_use_gpu=True):
554 with testing_utils.device(should_use_gpu=False):
561 with testing_utils.device(should_use_gpu=True):
571 with testing_utils.device(should_use_gpu=True):
787 with testing_utils.device(should_use_gpu=True):
794 with testing_utils.device(should_use_gpu=True):
/external/tensorflow/tensorflow/python/keras/
Dtesting_utils.py952 def device(should_use_gpu): argument
954 if should_use_gpu and test_util.is_gpu_available():
965 with device(should_use_gpu=True):