Searched refs:CanOpRunOnCPUDevice (Results 1 – 1 of 1) sorted by relevance
799 bool CanOpRunOnCPUDevice(const Node* n) { in CanOpRunOnCPUDevice() function in tensorflow::MklLayoutRewritePass3578 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()