Searched refs:ConvFwdPd (Results 1 – 4 of 4) sorted by relevance
53 using ConvFwdPd = mkldnn::convolution_forward::primitive_desc; typedef167 std::shared_ptr<ConvFwdPd> GetPrimitiveDesc() const { in GetPrimitiveDesc()190 std::shared_ptr<ConvFwdPd> fwd_pd;247 context_.fwd_pd.reset(new ConvFwdPd(*context_.fwd_desc, cpu_engine_)); in Setup()280 new ConvFwdPd(*context_.fwd_desc, post_ops_attr, cpu_engine_)); in Setup()282 context_.fwd_pd.reset(new ConvFwdPd(*context_.fwd_desc, cpu_engine_)); in Setup()665 std::shared_ptr<ConvFwdPd> conv_fwd_pd = conv_fwd->GetPrimitiveDesc(); in Compute()839 std::shared_ptr<ConvFwdPd>& conv2d_fwd_pd, in GetBiasHandle()849 const ConvFwdPd& conv_prim_desc, in AllocateOutputTensor()967 const ConvFwdPd& conv_prim_desc) const { in GetFilterTfDataFormat()[all …]
152 std::shared_ptr<ConvFwdPd> fwd_pd;214 context_.fwd_pd.reset(new ConvFwdPd(*context_.fwd_desc, cpu_engine_)); in Setup()
173 std::shared_ptr<ConvFwdPd> fwd_pd;236 context_.fwd_pd.reset(new ConvFwdPd(*context_.fwd_desc, cpu_engine_)); in Setup()
54 using ConvFwdPd = mkldnn::convolution_forward::primitive_desc; variable