Function IsCpuCompatibleConv2D

tensorflow/core/grappler/optimizers/remapper.cc:300–304

}

bool IsCpuCompatibleConv2D(const NodeDef* conv2d) {
  DCHECK(IsConv2D(*conv2d)) << "Expected Conv2D op";
  return NodeIsOnCpu(conv2d) && IsCpuCompatibleDataType(conv2d) &&
         IsCpuCompatibleDataFormat(conv2d);
}

bool IsGpuCompatibleConv2D(const RemapperContext& ctx, const NodeDef* conv2d) {
    DCHECK(IsConv2D(*conv2d)) << "Expected Conv2D op";