提交 198811c6 编写于 作者: Y Yangzihao Wang 提交者: TensorFlower Gardener

Remove the dependency of cudnn_disable_conv_1x1_optimization flag of filter...

Remove the dependency of cudnn_disable_conv_1x1_optimization flag of filter with the same size as input shape.

PiperOrigin-RevId: 165461092
上级 95e07ea9
......@@ -598,8 +598,7 @@ class Conv2DSlowBackpropFilterOp : public OpKernel {
", n=", n, ", k=", k));
}
return;
} else if (!cudnn_disable_conv_1x1_optimization_ &&
dims.spatial_dims[0].filter_size ==
} else if (dims.spatial_dims[0].filter_size ==
dims.spatial_dims[0].input_size &&
dims.spatial_dims[1].filter_size ==
dims.spatial_dims[1].input_size &&
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册