Searched refs:is_gpu_available (Results 1 – 25 of 128) sorted by relevance
123456
280 if test.is_gpu_available(cuda_only=True):310 if test.is_gpu_available(cuda_only=True):346 if test.is_gpu_available(cuda_only=True):381 if test.is_gpu_available(cuda_only=True):416 if test.is_gpu_available(cuda_only=True):446 if test.is_gpu_available(cuda_only=True):477 if test.is_gpu_available(cuda_only=True):508 if test.is_gpu_available(cuda_only=True):539 if test.is_gpu_available(cuda_only=True):570 if test.is_gpu_available(cuda_only=True):[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):309 with self.session(use_gpu=test.is_gpu_available()):336 with self.session(use_gpu=test.is_gpu_available()):363 with self.session(use_gpu=test.is_gpu_available()):
71 err_tolerance = 3e-2 if test.is_gpu_available() else 2e-379 if test.is_gpu_available(cuda_only=True):
158 if test.is_gpu_available(cuda_only=True):334 if gpu_only and not test.is_gpu_available(cuda_only=True):910 if use_gpu and not test.is_gpu_available(cuda_only=True):1379 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():1394 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():1409 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():1424 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():1439 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():1454 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():1469 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():[all …]
70 if not test.is_gpu_available():77 if not test.is_gpu_available(cuda_only=True):87 if not test.is_gpu_available(cuda_only=True):236 if not test.is_gpu_available():300 if not test.is_gpu_available():424 if not test.is_gpu_available():612 if not test.is_gpu_available():
46 if use_gpu and test.is_gpu_available():65 if test.is_gpu_available(cuda_only=True):69 if not test.is_gpu_available(cuda_only=True):76 if test.is_gpu_available(908 if test.is_gpu_available():1072 if not test.is_gpu_available(cuda_only=True):1797 if not test.is_gpu_available():1884 if not test.is_gpu_available():2262 with self.session(use_gpu=test.is_gpu_available()) as sess:2264 if test.is_gpu_available():[all …]
46 with self.cached_session(use_gpu=test.is_gpu_available()):71 with self.cached_session(use_gpu=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):
58 if test.is_gpu_available():128 if test.is_gpu_available():317 if not test.is_gpu_available():342 if data_format == "NCHW" and not test.is_gpu_available():
426 if not test.is_gpu_available():439 if not test.is_gpu_available():471 data_formats = ["NHWC", "NCHW"] if test.is_gpu_available() else ["NHWC"]739 if not test.is_gpu_available():775 data_formats = ["NHWC", "NCHW"] if test.is_gpu_available() else ["NHWC"]849 if not test.is_gpu_available():885 data_formats = ["NHWC", "NCHW"] if test.is_gpu_available() else ["NHWC"]
45 if test_util.is_gpu_available():145 if test.is_gpu_available():322 if not test.is_gpu_available():342 if data_format == "NCHW" and not test.is_gpu_available():
362 with ops.device("/GPU:0" if test.is_gpu_available() else "/CPU:0"):592 with ops.device("/GPU:0" if test.is_gpu_available() else "/CPU:0"):874 with ops.device("/GPU:0" if test.is_gpu_available() else "/CPU:0"):917 with ops.device("/GPU:0" if test.is_gpu_available() else "/CPU:0"):930 with ops.device("/GPU:0" if test.is_gpu_available() else "/CPU:0"):969 if not test.is_gpu_available():1010 if not test.is_gpu_available():
142 if test.is_gpu_available(cuda_only=True):157 if test.is_gpu_available(cuda_only=True):173 if test.is_gpu_available(cuda_only=True):
101 if not test.is_gpu_available():158 if not test.is_gpu_available():
259 if not test_util.is_gpu_available():271 if not test_util.is_gpu_available():283 if not test_util.is_gpu_available():321 if not test_util.is_gpu_available():341 if not test_util.is_gpu_available():361 if not test_util.is_gpu_available():381 if not test_util.is_gpu_available():401 if not test_util.is_gpu_available():482 if not test_util.is_gpu_available():506 if not test_util.is_gpu_available():[all …]
144 if not test_util.is_gpu_available():156 if not test_util.is_gpu_available():194 if not test_util.is_gpu_available():217 if not test_util.is_gpu_available():
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):
101 if not requires_gpu or test.is_gpu_available(cuda_only=True):260 if not requires_gpu or test.is_gpu_available(cuda_only=True):375 if not requires_gpu or test.is_gpu_available(cuda_only=True):429 if test.is_gpu_available(cuda_only=True):461 if test.is_gpu_available(cuda_only=True):479 if test.is_gpu_available(cuda_only=True):522 test.is_gpu_available(cuda_only=True)):555 if 'data_format' not in kwargs or test.is_gpu_available(cuda_only=True):1175 if 'data_format' not in kwargs or test.is_gpu_available(cuda_only=True):
247 if not test_util.is_gpu_available():268 if not test_util.is_gpu_available():291 if not test_util.is_gpu_available() or context.executing_eagerly():362 if not test_util.is_gpu_available():378 if not test_util.is_gpu_available():401 if not test_util.is_gpu_available():440 if not test_util.is_gpu_available():455 if not test_util.is_gpu_available():
156 if not test_util.is_gpu_available():173 if not test_util.is_gpu_available():207 if not test_util.is_gpu_available():238 if not test_util.is_gpu_available():
179 if test_util.is_gpu_available():203 if test_util.is_gpu_available():224 if test_util.is_gpu_available():244 if test_util.is_gpu_available():259 if not test_util.is_gpu_available():289 if not test_util.is_gpu_available():367 if not gpu_compatible and test.is_gpu_available():
120 if not test.is_gpu_available(cuda_only=True):139 if not test.is_gpu_available(cuda_only=True):225 if not test.is_gpu_available():
30 return 'channels_first' if tf.test.is_gpu_available() else 'channels_last'73 dev = 'gpu' if tf.test.is_gpu_available() else 'cpu'
40 if test.is_gpu_available():
101 if not test.is_gpu_available():180 if test.is_gpu_available():