Searched refs:NodeIsOnGpu (Results 1 – 3 of 3) sorted by relevance
246 bool NodeIsOnGpu(const NodeDef* node);
225 bool NodeIsOnGpu(const NodeDef* node) { in NodeIsOnGpu() function
185 return NodeIsOnGpu(conv2d) && IsGpuCompatibleDataType(conv2d) && in IsGpuCompatibleConv2D()