@@ -121,7 +121,7 @@ void TransferLayoutGeneral(const Context& dev_ctx,
121121
122122#ifdef PADDLE_WITH_DNNL
123123template <typename Context>
124- void TransferLayoutMKLDNN (const Context& dev_ctx,
124+ void TransferLayoutOneDNN (const Context& dev_ctx,
125125 const DenseTensor& x,
126126 DataLayout src_layout,
127127 DataLayout dst_layout,
@@ -152,10 +152,10 @@ void TransferLayoutMKLDNN(const Context& dev_ctx,
152152 }
153153
154154 if (src_layout != DataLayout::ONEDNN && dst_layout == DataLayout::ONEDNN) {
155- // Case1 - transform from Non-MKLDNN OPKernel to OneDNN OPKernel
155+ // Case1 - transform from Non-OneDNN OPKernel to OneDNN OPKernel
156156 // Just set layout/format. No real transform occur
157157 out->ShareDataWith (x);
158- // For NHWC data we need reshape of tensors as MKL-DNN
158+ // For NHWC data we need reshape of tensors as OneDNN
159159 // is expecting NHWC dims description order
160160 if (src_layout == DataLayout::NHWC) {
161161 VLOG (4 ) << " NHWC" ;
@@ -167,7 +167,7 @@ void TransferLayoutMKLDNN(const Context& dev_ctx,
167167 out->set_mem_desc (out_mem_desc);
168168 } else if (src_layout == DataLayout::ONEDNN &&
169169 dst_layout != DataLayout::ONEDNN) {
170- // Case2 - transform from OneDNN OPKernel to Non-MKLDNN OPKernel
170+ // Case2 - transform from OneDNN OPKernel to Non-OneDNN OPKernel
171171 // Do transform via OneDNN lib
172172 funcs::TransDataLayoutFromOneDNN (
173173 src_layout, dst_layout, x, out, dev_ctx.GetPlace ());
@@ -205,7 +205,7 @@ void TransferLayoutKernel(const Context& dev_ctx,
205205 }
206206
207207#ifdef PADDLE_WITH_DNNL
208- TransferLayoutMKLDNN <Context>(dev_ctx,
208+ TransferLayoutOneDNN <Context>(dev_ctx,
209209 x,
210210 static_cast <DataLayout>(src_layout),
211211 static_cast <DataLayout>(dst_layout),
0 commit comments