Searched refs:pooling_forward (Results 1 – 5 of 5) sorted by relevance
35 using dnnl::pooling_forward;39 using PoolingFwdPd = dnnl::pooling_forward::primitive_desc;124 std::shared_ptr<dnnl::pooling_forward::desc> fwd_desc;132 std::shared_ptr<dnnl::pooling_forward> fwd;163 MklPoolingFwdPrimitive<T>* pooling_forward = nullptr; in Get() local166 pooling_forward = static_cast<MklPoolingFwdPrimitive<T>*>( in Get()170 if (pooling_forward == nullptr) { in Get()171 pooling_forward = new MklPoolingFwdPrimitive<T>(fwdParams); in Get()173 fwdParams, pooling_forward); in Get()175 return pooling_forward; in Get()[all …]
53 context_.fwd_desc.reset(new pooling_forward::desc( in Setup()58 new pooling_forward::primitive_desc(*context_.fwd_desc, cpu_engine_)); in Setup()75 context_.fwd.reset(new pooling_forward(*context_.fwd_pd)); in Setup()79 context_.fwd.reset(new pooling_forward(*context_.fwd_pd)); in Setup()157 context_.fwd_desc.reset(new pooling_forward::desc( in Setup()162 new pooling_forward::primitive_desc(*context_.fwd_desc, cpu_engine_)); in Setup()
30 using dnnl::pooling_forward;210 const pooling_forward::primitive_desc& pool_fwd_prim_desc, in AllocateWorkspaceTensor()
30 using dnnl::pooling_forward;
276 ideep::pooling_forward::compute( in _mkldnn_pooling()