Searched refs:is_gpu_available (Results 1 – 25 of 128) sorted by relevance
123456
259 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 …]
157 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 …]
158 with self.session(use_gpu=test.is_gpu_available()):170 with self.session(use_gpu=test.is_gpu_available()):196 with self.session(use_gpu=test.is_gpu_available()):222 with self.session(use_gpu=test.is_gpu_available()):249 if test.is_gpu_available(cuda_only=True):307 with self.session(use_gpu=test.is_gpu_available()):334 with self.session(use_gpu=test.is_gpu_available()):361 with self.session(use_gpu=test.is_gpu_available()):
74 if not test.is_gpu_available():81 if not test.is_gpu_available(cuda_only=True):91 if not test.is_gpu_available(cuda_only=True):249 if not test.is_gpu_available():310 if not test.is_gpu_available():421 if not test.is_gpu_available():613 if not test.is_gpu_available():
44 if test.is_gpu_available():114 if test.is_gpu_available():303 if not test.is_gpu_available():328 if data_format == "NCHW" and not test.is_gpu_available():
44 if test_util.is_gpu_available():144 if test.is_gpu_available():293 if not test.is_gpu_available():313 if data_format == "NCHW" and not test.is_gpu_available():
155 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):
45 if use_gpu and test.is_gpu_available():62 if not test.is_gpu_available(cuda_only=True):69 if test.is_gpu_available(755 if test.is_gpu_available():908 if not test.is_gpu_available(cuda_only=True):1439 if not test.is_gpu_available():1519 if not test.is_gpu_available():1886 with self.session(use_gpu=test.is_gpu_available()) as sess:1888 if test.is_gpu_available():1923 if not test.is_gpu_available(cuda_only=True):[all …]
44 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):
153 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):
184 if gpu_only and not test.is_gpu_available():345 if gpu_only and not test.is_gpu_available():361 if gpu_only and not test.is_gpu_available():373 if gpu_only and not test.is_gpu_available():385 if gpu_only and not test.is_gpu_available():399 if gpu_only and not test.is_gpu_available():411 if gpu_only and not test.is_gpu_available():423 if gpu_only and not test.is_gpu_available():435 if gpu_only and not test.is_gpu_available():447 if gpu_only and not test.is_gpu_available():[all …]
345 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 …]
35 if not googletest.is_gpu_available():49 if not googletest.is_gpu_available():
27 if test.is_gpu_available(cuda_only=True):
100 if not test.is_gpu_available():179 if test.is_gpu_available():
44 model = rnn_ptb.test_model(tf.test.is_gpu_available())109 if not tf.test.is_gpu_available():114 if not tf.test.is_gpu_available():154 if not tf.test.is_gpu_available():159 if not tf.test.is_gpu_available():
256 if not test_util.is_gpu_available():277 if not test_util.is_gpu_available():298 if not test_util.is_gpu_available():336 if not test_util.is_gpu_available():356 if not test_util.is_gpu_available():376 if not test_util.is_gpu_available():396 if not test_util.is_gpu_available():416 if not test_util.is_gpu_available():498 if not test_util.is_gpu_available():522 if not test_util.is_gpu_available():[all …]
57 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):
95 if test_util.is_gpu_available():121 if test_util.is_gpu_available():141 if test_util.is_gpu_available():163 if test_util.is_gpu_available():178 if not test_util.is_gpu_available():207 if not test_util.is_gpu_available():328 if not works_on_gpu and test.is_gpu_available():
39 data_format = ('channels_first') if tf.test.is_gpu_available() else (62 data_format = ('channels_first') if tf.test.is_gpu_available() else (85 data_format = ('channels_first') if tf.test.is_gpu_available() else (102 if tf.test.is_gpu_available():157 'channels_first') if tf.test.is_gpu_available() else ('/cpu:0',
36 if not test.is_gpu_available():54 if not test.is_gpu_available():
116 if not test.is_gpu_available(cuda_only=True):132 if not test.is_gpu_available(cuda_only=True):218 if not test.is_gpu_available():
94 if not tf.test.is_gpu_available():127 if not tf.test.is_gpu_available():192 if not tf.test.is_gpu_available():230 if not tf.test.is_gpu_available():
49 if tf.test.is_gpu_available():129 if tf.test.is_gpu_available():
36 return 'channels_first' if tf.test.is_gpu_available() else 'channels_last'80 dev = 'gpu' if tf.test.is_gpu_available() else 'cpu'