Searched refs:bwd_desc (Results 1 – 4 of 4) sorted by relevance
149 context_.bwd_desc.reset(new pooling_backward::desc( in Setup()161 *context_.bwd_desc, cpu_engine_, *context_.fwd_pd)); in Setup()
268 std::shared_ptr<mkldnn::pooling_backward::desc> bwd_desc; member289 bwd_desc(nullptr), in PoolingBwdContext()
294 std::shared_ptr<mkldnn::eltwise_backward::desc> bwd_desc; member347 context_.bwd_desc.reset(new mkldnn::eltwise_backward::desc( in Setup()351 new EltwiseBwdPd(*context_.bwd_desc, cpu_engine_, *context_.fwd_pd)); in Setup()
551 auto bwd_desc = batch_normalization_backward::desc( in Setup() local553 context_.bwd_pd.reset(new BatchNormBwdPd(bwd_desc, cpu_engine_, fwd_pd)); in Setup()