Searched defs:dst_pd (Results 1 – 7 of 7) sorted by relevance
196 memory::desc dst_pd = lrn_fwd_prim_desc.dst_desc(); in AllocateOutputTensor() local459 memory::desc dst_pd = lrn_bkwd_prim_desc.diff_src_desc(); in AllocateOutputTensor() local
303 auto dst_pd = softmax_fwd->GetSoftmaxFwdPd()->dst_desc(); in Compute() local
664 memory::desc dst_pd = pool_fwd_prim_desc.dst_desc(); in AllocateOutputTensor() local717 memory::desc dst_pd = pool_bkwd_prim_desc.diff_src_desc(); in AllocateOutputTensor() local
570 auto dst_pd = conv_pd.diff_src_desc(); in AllocateOutputTensor() local
731 auto dst_pd = concat_pd.dst_desc(); in Compute() local
502 auto dst_pd = eltwise_fwd_pd->dst_desc(); in Compute() local
910 auto dst_pd = bn_fwd->GetDstPd(); in Compute() local