Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_layout_pass.cc1058 bool CanOpRunOnCPUDevice(const Node* n) { in CanOpRunOnCPUDevice() function in tensorflow::MklLayoutRewritePass
4011 if (!n->IsOp() || !CanOpRunOnCPUDevice(n)) { in RunPass()
4016 if ((m = CheckForNodeMerge(n)) != nullptr && CanOpRunOnCPUDevice(m)) { in RunPass()
4039 if (!n->IsOp() || !CanOpRunOnCPUDevice(n)) { in RunPass()
4061 if (!n->IsOp() || !CanOpRunOnCPUDevice(n)) { in RunPass()
4089 if (!n->IsOp() || !CanOpRunOnCPUDevice(n)) { in RunPass()