Searched refs:IsGpuCompatibleConv2D (Results 1 – 1 of 1) sorted by relevance
285 bool IsGpuCompatibleConv2D(const NodeDef* conv2d) { in IsCpuCompatibleDataType() function354 return is_relu && is_spatial_conv && IsGpuCompatibleConv2D(&contraction_node); in IsCpuCompatibleDataType()