diff options
Diffstat (limited to 'tensorflow/core/kernels/conv_ops_3d.cc')
| -rw-r--r-- | tensorflow/core/kernels/conv_ops_3d.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/conv_ops_3d.cc b/tensorflow/core/kernels/conv_ops_3d.cc index 9ec16be67d..a1eed4e68c 100644 --- a/tensorflow/core/kernels/conv_ops_3d.cc +++ b/tensorflow/core/kernels/conv_ops_3d.cc @@ -415,6 +415,7 @@ struct LaunchConvOp<GPUDevice, T> { in_batch, in_depth, {{in_planes, in_rows, in_cols}}, + FORMAT_NCHW, out_depth, {{filter_planes, filter_rows, filter_cols}}, {{dilations[0], dilations[1], dilations[2]}}, |