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

Function IsCPUDevice

tensorflow/python/lib/core/py_func.cc:72–74  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

70};
71
72bool IsCPUDevice(const Device* d) {
73 return d == nullptr || d->tensorflow_gpu_device_info() == nullptr;
74}
75
76// Givens the 'call', prepares the token and inputs as a python tuple
77// that is appropriate for calling the trampoline.

Callers 2

MakeArgTupleFunction · 0.85

Calls 1

Tested by

no test coverage detected