Function IsGpuCompatibleConv2D

tensorflow/core/grappler/optimizers/remapper.cc:306–310

}

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

bool IsGpuCompatibleMatMul(const RemapperContext& ctx,
    const NodeDef* matmul) {