Searched refs:NodeIsOnGpu (Results 1 – 3 of 3) sorted by relevance
325 bool NodeIsOnGpu(const NodeDef* node);
160 bool NodeIsOnGpu(const NodeDef* node) { in NodeIsOnGpu() function
287 return NodeIsOnGpu(conv2d) && IsGpuCompatibleDataType(conv2d) && in IsCpuCompatibleDataType()848 if (!NodeIsOnGpu(fused_batch_norm_node_def)) return false; in IsCpuCompatibleDataType()868 if (is_training && NodeIsOnGpu(fused_batch_norm_node_def)) { in IsCpuCompatibleDataType()