Searched refs:ConvFwdPd (Results 1 – 1 of 1) sorted by relevance
96 typedef mkldnn::convolution_forward::primitive_desc ConvFwdPd; typedef165 std::shared_ptr<ConvFwdPd> GetPrimitiveDesc() const { in GetPrimitiveDesc()192 std::shared_ptr<ConvFwdPd> fwd_pd;244 context_.fwd_pd.reset(new ConvFwdPd(*context_.fwd_desc, cpu_engine_)); in Setup()276 new ConvFwdPd(*context_.fwd_desc, post_ops_attr, cpu_engine_)); in Setup()278 context_.fwd_pd.reset(new ConvFwdPd(*context_.fwd_desc, cpu_engine_)); in Setup()1084 std::shared_ptr<ConvFwdPd> conv_fwd_pd = conv_fwd->GetPrimitiveDesc(); in Compute()1240 std::shared_ptr<ConvFwdPd>& conv2d_fwd_pd, in GetBiasHandle()1250 const ConvFwdPd& conv_prim_desc, in AllocateOutputTensor()1306 const ConvFwdPd& conv_prim_desc, in AllocatePersistentTensor()[all …]