aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/kernels/conv_ops.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/kernels/conv_ops.cc')
-rw-r--r--tensorflow/core/kernels/conv_ops.cc7
1 files changed, 3 insertions, 4 deletions
diff --git a/tensorflow/core/kernels/conv_ops.cc b/tensorflow/core/kernels/conv_ops.cc
index eb3102f19d..df278d036a 100644
--- a/tensorflow/core/kernels/conv_ops.cc
+++ b/tensorflow/core/kernels/conv_ops.cc
@@ -219,12 +219,11 @@ class LaunchXsmmConvOp<CPUDevice, float> {
desc.pad_w_out = 0;
desc.threads = num_threads;
desc.algo = LIBXSMM_DNN_CONV_ALGO_DIRECT;
- desc.buffer_format = LIBXSMM_DNN_CONV_FORMAT_NHWC;
- desc.filter_format = LIBXSMM_DNN_CONV_FORMAT_LIBXSMM;
+ desc.buffer_format = LIBXSMM_DNN_TENSOR_FORMAT_NHWC;
+ desc.filter_format = LIBXSMM_DNN_TENSOR_FORMAT_LIBXSMM;
desc.fuse_ops = LIBXSMM_DNN_CONV_FUSE_NONE;
desc.options = LIBXSMM_DNN_CONV_OPTION_NONE;
- desc.datatype_in = LIBXSMM_DNN_DATATYPE_F32;
- desc.datatype_out = LIBXSMM_DNN_DATATYPE_F32;
+ desc.datatype = LIBXSMM_DNN_DATATYPE_F32;
if (!CanUseXsmmConv2D(desc, data_format)) {
return false;