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