HomeSort by relevance Sort by last modified time
    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