OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
full:is_gpu_available
(Results
1 - 25
of
88
) sorted by null
1
2
3
4
/external/tensorflow/tensorflow/contrib/eager/python/examples/rnn_ptb/
rnn_ptb_graph_test.py
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
():
153
if not tf.test.
is_gpu_available
():
158
if not tf.test.
is_gpu_available
():
/external/tensorflow/tensorflow/docs_src/api_guides/python/
test.md
40
* @{tf.test.
is_gpu_available
}
/external/tensorflow/tensorflow/python/ops/
nn_fused_batchnorm_test.py
344
if test.
is_gpu_available
(cuda_only=True):
354
if test.
is_gpu_available
(cuda_only=True):
363
if test.
is_gpu_available
(cuda_only=True):
371
if test.
is_gpu_available
(cuda_only=True):
382
if test.
is_gpu_available
(cuda_only=True):
393
if test.
is_gpu_available
(cuda_only=True):
404
if test.
is_gpu_available
(cuda_only=True):
412
if test.
is_gpu_available
(cuda_only=True):
420
if test.
is_gpu_available
(cuda_only=True):
431
if test.
is_gpu_available
(cuda_only=True)
[
all
...]
spectral_ops_test_util.py
27
if test.
is_gpu_available
(cuda_only=True):
matmul_benchmark_test.py
35
if not googletest.
is_gpu_available
():
49
if not googletest.
is_gpu_available
():
/external/tensorflow/tensorflow/tools/api/golden/
tensorflow.test.pbtxt
48
name: "
is_gpu_available
"
/external/tensorflow/tensorflow/python/grappler/
layout_optimizer_test.py
254
if test.
is_gpu_available
(cuda_only=True):
283
if test.
is_gpu_available
(cuda_only=True):
318
if test.
is_gpu_available
(cuda_only=True):
352
if test.
is_gpu_available
(cuda_only=True):
386
if test.
is_gpu_available
(cuda_only=True):
415
if test.
is_gpu_available
(cuda_only=True):
445
if test.
is_gpu_available
(cuda_only=True):
475
if test.
is_gpu_available
(cuda_only=True):
505
if test.
is_gpu_available
(cuda_only=True):
535
if test.
is_gpu_available
(cuda_only=True)
[
all
...]
/external/tensorflow/tensorflow/contrib/memory_stats/python/kernel_tests/
memory_stats_ops_test.py
36
if not test.
is_gpu_available
():
54
if not test.
is_gpu_available
():
/external/tensorflow/tensorflow/python/kernel_tests/
pool_test.py
157
with self.test_session(use_gpu=test.
is_gpu_available
()):
169
with self.test_session(use_gpu=test.
is_gpu_available
()):
195
with self.test_session(use_gpu=test.
is_gpu_available
()):
221
with self.test_session(use_gpu=test.
is_gpu_available
()):
248
if test.
is_gpu_available
(cuda_only=True):
305
with self.test_session(use_gpu=test.
is_gpu_available
()):
331
with self.test_session(use_gpu=test.
is_gpu_available
()):
357
with self.test_session(use_gpu=test.
is_gpu_available
()):
scatter_ops_test.py
145
if not test.
is_gpu_available
():
162
if not test.
is_gpu_available
():
183
if test.
is_gpu_available
():
lrn_op_test.py
99
if not test.
is_gpu_available
():
154
if not test.
is_gpu_available
():
conv_ops_test.py
152
if test.
is_gpu_available
(cuda_only=True):
[
all
...]
conv2d_backprop_filter_grad_test.py
68
if test.
is_gpu_available
(cuda_only=True):
/external/tensorflow/tensorflow/python/platform/
test.py
29
@@
is_gpu_available
52
from tensorflow.python.framework.test_util import
is_gpu_available
/external/tensorflow/tensorflow/python/client/
device_lib_test.py
35
if test.
is_gpu_available
():
virtual_gpu_test.py
204
# TODO(laigd): b/70811538. The
is_gpu_available
() call will invoke
209
if not test.
is_gpu_available
(cuda_only=True):
237
if not test.
is_gpu_available
(cuda_only=True):
/external/tensorflow/tensorflow/contrib/fused_conv/python/ops/
fused_conv2d_bias_activation_op_test.py
302
if gpu_only and not test.
is_gpu_available
():
318
if gpu_only and not test.
is_gpu_available
():
330
if gpu_only and not test.
is_gpu_available
():
342
if gpu_only and not test.
is_gpu_available
():
356
if gpu_only and not test.
is_gpu_available
():
368
if gpu_only and not test.
is_gpu_available
():
380
if gpu_only and not test.
is_gpu_available
():
392
if gpu_only and not test.
is_gpu_available
():
404
if gpu_only and not test.
is_gpu_available
():
432
if gpu_only and not test.
is_gpu_available
()
[
all
...]
/external/tensorflow/tensorflow/python/profiler/internal/
run_metadata_test.py
110
if not test.
is_gpu_available
(cuda_only=True):
125
if not test.
is_gpu_available
(cuda_only=True):
209
if not test.
is_gpu_available
():
/external/tensorflow/tensorflow/compiler/tests/
xla_device_test.py
32
if not test.
is_gpu_available
():
/external/tensorflow/tensorflow/contrib/eager/python/examples/linear_regression/
linear_regression_graph_test.py
44
if tf.test.
is_gpu_available
():
/external/tensorflow/tensorflow/contrib/eager/python/examples/notebooks/
1_basics.ipynb
425
"
is_gpu_available
= tfe.num_gpus() \u003e 0\n",
431
"if
is_gpu_available
:\n",
475
"if
is_gpu_available
:\n",
508
"if
is_gpu_available
:\n",
/external/tensorflow/tensorflow/contrib/eager/python/examples/gan/
mnist_graph_test.py
36
return 'channels_first' if tf.test.
is_gpu_available
() else 'channels_last'
79
dev = 'gpu' if tf.test.
is_gpu_available
() else 'cpu'
mnist_test.py
36
return 'channels_first' if tf.test.
is_gpu_available
() else 'channels_last'
/external/tensorflow/tensorflow/contrib/eager/python/examples/resnet50/
resnet50_graph_test.py
32
return 'channels_first' if tf.test.
is_gpu_available
() else 'channels_last'
107
dev = 'gpu' if tf.test.
is_gpu_available
() else 'cpu'
/external/tensorflow/tensorflow/contrib/eager/python/examples/mnist/
mnist_graph_test.py
26
return "channels_first" if tf.test.
is_gpu_available
() else "channels_last"
Completed in 657 milliseconds
1
2
3
4