message: Use buffer as key
diff: @@ -24,8 +24,8 @@ limitations under the License.
  #include <map>
  #include <vector>
  -#include "mkldnn.hpp"
  #include "absl/strings/str_join.h"
  +#include "mkldnn.hpp"
  #include "tensorflow/core/framework/bounds_check.h"
  #include "tensorflow/core/framework/numeric_op.h"
  #include "tensorflow/core/framework/op_kernel.h"
  @@ -576,15 +576,13 @@ class MklConvOp : public OpKernel {
  OP_REQUIRES(context, dilations_.size() == 5,
  errors::InvalidArgument("Dilation rates field must "
  "specify 5 dimensions"));
  - OP_REQUIRES(context,
  - (GetTensorDim(dilations_, data_format_, 'N') == 1 &&
  + OP_REQUIRES(context, (GetTensorDim(dilations_, data_format_, 'N') == 1 &&
  GetTensorDim(dilations_, data_format_, 'C') == 1),
  errors::InvalidArgument(
  "Current implementation does not yet support "
  "dilations rates in the batch and depth dimensions."));
  OP_REQUIRES(
  - context,
  - (GetTensorDim(dilations_, data_format_, '0') > 0 &&
  + context, (GetTensorDim(dilations_, data_format_, '0') > 0 &&
  GetTensorDim(dilations_, data_format_, '1') > 0 &&
  GetTensorDim(dilations_, data_format_, '2') > 0),
  errors::InvalidArgument("Dilated rates should be larger than 0."));
  @@ -1579,8 +1577,8 @@ class MklQuantizedConv2DOp
  param_key.AddAsKey<float>(max_input);
  param_key.AddAsKey<float>(min_freezed_output);
  param_key.AddAsKey<float>(max_freezed_output);
  - param_key.AddAsKey<const Tensor*>(&min_filter_vector);
  - param_key.AddAsKey<const Tensor*>(&max_filter_vector);
  + param_key.AddAsKey<const float*>(min_filter);
  + param_key.AddAsKey<const float*>(max_filter);
  params.post_op_params.push_back(
  {"output_scale", ALGORITHM_UNDEF, scales, param_key.GetKey()});
  }
  