Searched refs:BatchNormBwdPd (Results 1 – 1 of 1) sorted by relevance
36 using BatchNormBwdPd = mkldnn::batch_normalization_backward::primitive_desc; typedef480 std::shared_ptr<BatchNormBwdPd> GetBatchNormBwdPd() const { in GetBatchNormBwdPd()501 std::shared_ptr<BatchNormBwdPd> bwd_pd;553 context_.bwd_pd.reset(new BatchNormBwdPd(bwd_desc, cpu_engine_, fwd_pd)); in Setup()1226 std::shared_ptr<BatchNormBwdPd> bn_bwd_pd = bn_bwd->GetBatchNormBwdPd(); in Compute()