diff options
Diffstat (limited to 'tensorflow/core/kernels/mkl_conv_ops.h')
| -rw-r--r-- | tensorflow/core/kernels/mkl_conv_ops.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/mkl_conv_ops.h b/tensorflow/core/kernels/mkl_conv_ops.h index 47a9b4bfc7..b6883dbaa2 100644 --- a/tensorflow/core/kernels/mkl_conv_ops.h +++ b/tensorflow/core/kernels/mkl_conv_ops.h @@ -18,6 +18,7 @@ limitations under the License. #include <vector> #include <limits> +#include <string> #include "tensorflow/core/framework/numeric_op.h" #include "tensorflow/core/framework/op_kernel.h" @@ -288,7 +289,7 @@ class MklDnnConvUtil { OP_REQUIRES(context_, input_tf_shape.dims() == 4, errors::InvalidArgument("input must be 4-dimensional", - input_tf_shape.DebugString())); + input_tf_shape.DebugString())); GetOutputAndPadSizeInMklOrder(input_tf_shape, filter_tf_shape, strides, output_dims_tf_order, |