Searched refs:NodeIsOnCpu (Results 1 – 4 of 4) sorted by relevance
322 bool NodeIsOnCpu(const NodeDef* node);
154 bool NodeIsOnCpu(const NodeDef* node) { in NodeIsOnCpu() function
281 return NodeIsOnCpu(conv2d) && IsCpuCompatibleDataType(conv2d) && in IsCpuCompatibleDataType()293 return NodeIsOnCpu(matmul) && IsCpuCompatibleDataType(matmul); in IsCpuCompatibleDataType()299 return NodeIsOnCpu(dw_conv2d) && IsCpuCompatibleDataType(dw_conv2d); in IsCpuCompatibleDataType()576 if (!IsConv2D(*conv2d_node_def) || !NodeIsOnCpu(conv2d_node_def) || in IsCpuCompatibleDataType()
194 bool NodeIsOnCpu(const NodeDef& node) { in NodeIsOnCpu() function2444 if (!is_complex || NodeIsOnCpu(*node)) { in TrySimplify()2607 if (NodeIsOnCpu(*node)) { in TrySimplify()3096 if (!NodeIsOnCpu(node)) { in CanOptimize()