cleaing up comments

1 files changed, 2 insertions, 2 deletions

diff --git a/tensorflow/core/kernels/mkl_softmax_op.cc b/tensorflow/core/kernels/mkl_softmax_op.cc
index 1a01ee37be..ae286fcd28 100644
--- a/tensorflow/core/kernels/mkl_softmax_op.cc
+++ b/tensorflow/core/kernels/mkl_softmax_op.cc
@@ -81,7 +81,7 @@ class MklSoftmaxOp : public OpKernel {
           layout_type = memory::format::ncdhw;
           break;
         default:
-          OP_REQUIRES_OK(context, errors::Aborted("Input dims must be <= 5:"));
+          OP_REQUIRES_OK(context, errors::Aborted("Input dims must be <= 5 and >=1"));
       }
       // Create softmax memory for src, dst: both are defined in mkl_util.h,
       // they are wrapper
@@ -107,7 +107,7 @@ class MklSoftmaxOp : public OpKernel {
       src.SetOpMemDesc(src_dims, layout_type);
 
       // creating a memory descriptor
-      // passing outermost dim as default axis
+      // passing outermost dim as default axis, where the softmax is applied
       int axis = input_dims - 1;
       auto softmax_fwd_desc = softmax_forward::desc(prop_kind::forward_scoring,
                                                     src.GetOpMemDesc(), axis);