Searched refs:NodeIsOnCpu (Results 1 – 4 of 4) sorted by relevance
243 bool NodeIsOnCpu(const NodeDef* node);
219 bool NodeIsOnCpu(const NodeDef* node) { in NodeIsOnCpu() function
179 return NodeIsOnCpu(conv2d) && IsCpuCompatibleDataType(conv2d) && in IsCpuCompatibleConv2D()372 !NodeIsOnCpu(conv2d.node) || // in FindConv2DWithBatchNorm()
2985 if (!NodeIsOnCpu(node)) { in CanOptimize()2996 bool NodeIsOnCpu(const NodeDef& node) const { in NodeIsOnCpu() function in tensorflow::grappler::__anonbb66c6470111::UnaryOpsComposition