@@ -222,35 +222,35 @@ struct MKLDNNActivationGradFunc : public BaseActivationFunctor<T> {
222
222
};
223
223
224
224
template <typename T>
225
- using ReluMkldnnFunctor =
225
+ using ReluMKLDNNFunctor =
226
226
MKLDNNActivationFunc<T, mkldnn::algorithm::eltwise_relu>;
227
227
228
228
template <typename T>
229
- using TanhMkldnnFunctor =
229
+ using TanhMKLDNNFunctor =
230
230
MKLDNNActivationFunc<T, mkldnn::algorithm::eltwise_tanh>;
231
231
232
232
template <typename T>
233
- using SqrtMkldnnFunctor =
233
+ using SqrtMKLDNNFunctor =
234
234
MKLDNNActivationFunc<T, mkldnn::algorithm::eltwise_sqrt>;
235
235
236
236
template <typename T>
237
- using AbsMkldnnFunctor =
237
+ using AbsMKLDNNFunctor =
238
238
MKLDNNActivationFunc<T, mkldnn::algorithm::eltwise_abs>;
239
239
240
240
template <typename T>
241
- using ReluMkldnnGradFunctor =
241
+ using ReluMKLDNNGradFunctor =
242
242
MKLDNNActivationGradFunc<T, mkldnn::algorithm::eltwise_relu>;
243
243
244
244
template <typename T>
245
- using TanhMkldnnGradFunctor =
245
+ using TanhMKLDNNGradFunctor =
246
246
MKLDNNActivationGradFunc<T, mkldnn::algorithm::eltwise_tanh>;
247
247
248
248
template <typename T>
249
- using SqrtMkldnnGradFunctor =
249
+ using SqrtMKLDNNGradFunctor =
250
250
MKLDNNActivationGradFunc<T, mkldnn::algorithm::eltwise_sqrt>;
251
251
252
252
template <typename T>
253
- using AbsMkldnnGradFunctor =
253
+ using AbsMKLDNNGradFunctor =
254
254
MKLDNNActivationGradFunc<T, mkldnn::algorithm::eltwise_abs>;
255
255
} // namespace operators
256
256
} // namespace paddle
@@ -265,9 +265,9 @@ namespace ops = paddle::operators;
265
265
ops::MKLDNNActivationGradKernel<ops::grad_functor<float >>);
266
266
267
267
#define FOR_EACH_MKLDNN_KERNEL_FUNCTOR (__macro ) \
268
- __macro (relu, ReluMkldnnFunctor, ReluMkldnnGradFunctor ); \
269
- __macro (tanh, TanhMkldnnFunctor, TanhMkldnnGradFunctor ); \
270
- __macro (sqrt, SqrtMkldnnFunctor, SqrtMkldnnGradFunctor ); \
271
- __macro (abs, AbsMkldnnFunctor, AbsMkldnnGradFunctor );
268
+ __macro (relu, ReluMKLDNNFunctor, ReluMKLDNNGradFunctor ); \
269
+ __macro (tanh, TanhMKLDNNFunctor, TanhMKLDNNGradFunctor ); \
270
+ __macro (sqrt, SqrtMKLDNNFunctor, SqrtMKLDNNGradFunctor ); \
271
+ __macro (abs, AbsMKLDNNFunctor, AbsMKLDNNGradFunctor );
272
272
273
273
FOR_EACH_MKLDNN_KERNEL_FUNCTOR (REGISTER_ACTIVATION_MKLDNN_KERNEL);
0 commit comments