Searched refs:BatchNormFwdPd (Results 1 – 1 of 1) sorted by relevance
35 using BatchNormFwdPd = mkldnn::batch_normalization_forward::primitive_desc; typedef143 std::shared_ptr<BatchNormFwdPd> GetBatchNormFwdPd() const { in GetBatchNormFwdPd()165 std::shared_ptr<BatchNormFwdPd> fwd_pd;203 context_.fwd_pd.reset(new BatchNormFwdPd(fwd_desc, cpu_engine_)); in Setup()544 auto fwd_pd = BatchNormFwdPd(fwd_desc, cpu_engine_); in Setup()836 std::shared_ptr<BatchNormFwdPd> bn_fwd_pd = bn_fwd->GetBatchNormFwdPd(); in Compute()