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

Function IsCPU

tensorflow/c/eager/c_api.cc:90–92  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

88}
89
90bool IsCPU(const tensorflow::Device* d) {
91 return d == nullptr || d->tensorflow_gpu_device_info() == nullptr;
92}
93
94string DeviceName(const tensorflow::Device* d) {
95 return (d == nullptr) ? "cpu:0" : d->name();

Callers 1

TFE_TensorHandleResolveFunction · 0.85

Calls 1

Tested by

no test coverage detected