Searched defs:MklBatchNormFwdParams (Results 1 – 1 of 1) sorted by relevance
43 struct MklBatchNormFwdParams { struct44 memory::dims src_dims;45 int depth;46 float eps;47 bool training;48 FusedBNActivationMode activation_mode;49 memory::desc src_md;51 MklBatchNormFwdParams(const memory::dims& src_dims, int depth, float eps, in MklBatchNormFwdParams() argument