Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/graph/
Dmkl_layout_pass.cc799 bool CanOpRunOnCPUDevice(const Node* n) { in CanOpRunOnCPUDevice() function in tensorflow::MklLayoutRewritePass
3578 if (!n->IsOp() || !CanOpRunOnCPUDevice(n)) { in RunPass()
3583 if ((m = CheckForNodeMerge(n)) != nullptr && CanOpRunOnCPUDevice(m)) { in RunPass()
3606 if (!n->IsOp() || !CanOpRunOnCPUDevice(n)) { in RunPass()
3628 if (!n->IsOp() || !CanOpRunOnCPUDevice(n)) { in RunPass()
3656 if (!n->IsOp() || !CanOpRunOnCPUDevice(n)) { in RunPass()