Home
last modified time | relevance | path

Searched refs:bwd_pd (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_pooling_ops_common.cc160 context_.bwd_pd.reset(new pooling_backward::primitive_desc( in Setup()
164 context_.diff_src_mem.reset(new memory(context_.bwd_pd.get()->diff_src_desc(), in Setup()
166 context_.diff_dst_mem.reset(new memory(context_.bwd_pd.get()->diff_dst_desc(), in Setup()
177 context_.bwd.reset(new pooling_backward(*context_.bwd_pd)); in Setup()
182 context_.bwd.reset(new pooling_backward(*context_.bwd_pd)); in Setup()
Dmkl_relu_op.cc283 std::shared_ptr<EltwiseBwdPd> GetEltwiseBwdPd() { return context_.bwd_pd; } in GetEltwiseBwdPd()
310 std::shared_ptr<EltwiseBwdPd> bwd_pd; member
330 bwd_pd(nullptr), in EltwiseBwdContext()
350 context_.bwd_pd.reset( in Setup()
353 auto bwd_pd = context_.bwd_pd.get(); in Setup() local
357 new memory(bwd_pd->src_desc(), cpu_engine_, DummyData)); in Setup()
359 new memory(bwd_pd->diff_dst_desc(), cpu_engine_, DummyData)); in Setup()
361 new memory(bwd_pd->diff_src_desc(), cpu_engine_, DummyData)); in Setup()
363 context_.eltwise_bwd.reset(new mkldnn::eltwise_backward(*context_.bwd_pd)); in Setup()
Dmkl_pooling_ops_common.h233 return context_.bwd_pd; in GetPoolingBwdPd()
270 std::shared_ptr<PoolingBwdPd> bwd_pd; member
291 bwd_pd(nullptr), in PoolingBwdContext()
Dmkl_fused_batch_norm_op.cc481 return context_.bwd_pd; in GetBatchNormBwdPd()
501 std::shared_ptr<BatchNormBwdPd> bwd_pd; member
553 context_.bwd_pd.reset(new BatchNormBwdPd(bwd_desc, cpu_engine_, fwd_pd)); in Setup()
568 context_.bn_bwd.reset(new batch_normalization_backward(*context_.bwd_pd)); in Setup()