Searched refs:MklBatchNormFwdParams (Results 1 – 1 of 1) sorted by relevance
43 struct MklBatchNormFwdParams { struct51 MklBatchNormFwdParams(const memory::dims& src_dims, int depth, float eps, in MklBatchNormFwdParams() function65 explicit MklFusedBatchNormFwdPrimitive(const MklBatchNormFwdParams& fwdParams) in MklFusedBatchNormFwdPrimitive()185 void Setup(const MklBatchNormFwdParams& fwdParams) { in Setup()329 const MklBatchNormFwdParams& fwdParams) { in Get()351 static string CreateKey(const MklBatchNormFwdParams& fwdParams) { in CreateKey()365 MklPrimitive* GetBatchNormFwd(const MklBatchNormFwdParams& fwdParams) { in GetBatchNormFwd()370 void SetBatchNormFwd(const MklBatchNormFwdParams& fwdParams, in SetBatchNormFwd()776 MklBatchNormFwdParams fwdParams(src_dims, depth_, epsilon_, is_training_, in Compute()