diff options
Diffstat (limited to 'tensorflow/core/kernels/conv_grad_ops_3d.cc')
| -rw-r--r-- | tensorflow/core/kernels/conv_grad_ops_3d.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/conv_grad_ops_3d.cc b/tensorflow/core/kernels/conv_grad_ops_3d.cc index 980b1063de..15f1bf9aba 100644 --- a/tensorflow/core/kernels/conv_grad_ops_3d.cc +++ b/tensorflow/core/kernels/conv_grad_ops_3d.cc @@ -716,6 +716,7 @@ class Conv3DBackpropInputOp<GPUDevice, T> : public OpKernel { batch, in_depth, {{input_size[0], input_size[1], input_size[2]}}, + FORMAT_NCHW, out_depth, {{filter_size[0], filter_size[1], filter_size[2]}}, {{dilations[0], dilations[1], dilations[2]}}, @@ -1112,6 +1113,7 @@ class Conv3DBackpropFilterOp<GPUDevice, T> : public OpKernel { batch, in_depth, {{input_size[0], input_size[1], input_size[2]}}, + FORMAT_NCHW, out_depth, {{filter_size[0], filter_size[1], filter_size[2]}}, {{dilations[0], dilations[1], dilations[2]}}, |