Searched refs:MklBatchNormBwdParams (Results 1 – 1 of 1) sorted by relevance
377 struct MklBatchNormBwdParams { struct387 MklBatchNormBwdParams(memory::dims src_dims, memory::dims diff_dst_dims, in MklBatchNormBwdParams() argument402 explicit MklFusedBatchNormBwdPrimitive(const MklBatchNormBwdParams& bwdParams) in MklFusedBatchNormBwdPrimitive()519 void Setup(const MklBatchNormBwdParams& bwdParams) { in Setup()587 const MklBatchNormBwdParams& bwdParams) { in Get()608 static string CreateKey(const MklBatchNormBwdParams& bwdParams) { in CreateKey()622 MklPrimitive* GetBatchNormBwd(const MklBatchNormBwdParams& bwdParams) { in GetBatchNormBwd()627 void SetBatchNormBwd(const MklBatchNormBwdParams& bwdParams, in SetBatchNormBwd()1220 MklBatchNormBwdParams bwdParams(src_dims, diff_dst_dims, depth_, epsilon_, in Compute()