MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / IsGoogleCudaEnabled

Function IsGoogleCudaEnabled

tensorflow/core/util/port.cc:21–27  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

19namespace tensorflow {
20
21bool IsGoogleCudaEnabled() {
22#if GOOGLE_CUDA
23 return true;
24#else
25 return false;
26#endif
27}
28
29bool IsBuiltWithROCm() {
30#if TENSORFLOW_USE_ROCM

Callers 6

RunComposedOpMethod · 0.50
RunAutoClusteringTestImplMethod · 0.50
TESTFunction · 0.50
GetDevicesForNodeMethod · 0.50
BM_ConvFloatFunction · 0.50
BM_ConvFloatDepthwiseFunction · 0.50

Calls

no outgoing calls

Tested by 5

RunComposedOpMethod · 0.40
RunAutoClusteringTestImplMethod · 0.40
TESTFunction · 0.40
BM_ConvFloatFunction · 0.40
BM_ConvFloatDepthwiseFunction · 0.40