diff --git a/paddle/fluid/operators/mkldnn/transpose_mkldnn_op.cc b/paddle/fluid/operators/mkldnn/transpose_mkldnn_op.cc index 95cee806ac451235a8fb03567e6057e10aa56427..480167f43525bc19defab0faab31460e6c179eff 100644 --- a/paddle/fluid/operators/mkldnn/transpose_mkldnn_op.cc +++ b/paddle/fluid/operators/mkldnn/transpose_mkldnn_op.cc @@ -46,7 +46,8 @@ class TransposeMKLDNNOpKernel : public paddle::framework::OpKernel { std::vector nchw_tz = paddle::framework::vectorize2int(input->dims()); const std::string key = platform::TransposeMKLDNNHandler::GetHash( - nchw_tz, axis, ctx.op().Output("Out")); + nchw_tz, axis, + ctx.op().Output("Out") + std::to_string(input->format())); platform::TransposeMKLDNNHandler handler(nchw_tz, axis, dev_ctx, mkldnn_engine, key);