Home
last modified time | relevance | path

Searched refs:IsGoogleCudaEnabled (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dvariant_op_copy_test.cc197 if (!IsGoogleCudaEnabled()) return; in TEST()
240 if (!IsGoogleCudaEnabled()) return; in TEST()
317 if (!IsGoogleCudaEnabled()) return; in TEST()
354 if (!IsGoogleCudaEnabled()) return; in TEST()
/external/tensorflow/tensorflow/core/util/
Dport.h22 bool IsGoogleCudaEnabled();
Dport.cc21 bool IsGoogleCudaEnabled() { in IsGoogleCudaEnabled() function
/external/tensorflow/tensorflow/python/util/
Dport.i24 %unignore tensorflow::IsGoogleCudaEnabled;
/external/tensorflow/tensorflow/python/platform/
Dtest.py96 return _test_util.IsGoogleCudaEnabled()
/external/tensorflow/tensorflow/core/common_runtime/
Dcolocation_graph.cc768 if (!IsGoogleCudaEnabled() && in GetDevicesForNode()
/external/tensorflow/tensorflow/python/framework/
Dtest_util.py251 def IsGoogleCudaEnabled(): function
252 return pywrap_tensorflow.IsGoogleCudaEnabled()
Dtest_util_test.py111 if test_util.IsGoogleCudaEnabled():
/external/tensorflow/tensorflow/core/kernels/
Dnn_ops_test.cc110 if (!IsGoogleCudaEnabled() && use_gpu) { in BM_ConvFloat()
504 if (!IsGoogleCudaEnabled() && use_gpu) { in BM_ConvFloatDepthwise()