Searched refs:dst_pd (Results 1 – 8 of 8) sorted by relevance
280 auto dst_pd = softmax_fwd->GetSoftmaxFwdPd()->dst_desc(); in Compute() local286 output_mkl_shape.SetMklLayout(&dst_pd); in Compute()289 output_tf_shape.AddDim((dst_pd.get_size() / sizeof(T))); in Compute()
195 memory::desc dst_pd = lrn_fwd_prim_desc.dst_desc(); in AllocateOutputTensor() local201 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()207 size_t num_bytes = dst_pd.get_size(); in AllocateOutputTensor()458 memory::desc dst_pd = lrn_bkwd_prim_desc.diff_src_desc(); in AllocateOutputTensor() local463 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()469 size_t num_bytes = dst_pd.get_size(); in AllocateOutputTensor()
649 memory::desc dst_pd = pool_fwd_prim_desc.dst_desc(); in AllocateOutputTensor() local653 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()658 output_tf_shape.AddDim(this->GetNumTElements(dst_pd)); in AllocateOutputTensor()702 memory::desc dst_pd = pool_bkwd_prim_desc.diff_src_desc(); in AllocateOutputTensor() local705 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()711 output_tf_shape.AddDim(this->GetNumTElements(dst_pd)); in AllocateOutputTensor()
567 auto dst_pd = conv_pd.diff_src_desc(); in AllocateOutputTensor() local572 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()579 output_tf_shape.AddDim(dst_pd.get_size() / sizeof(T)); in AllocateOutputTensor()
395 auto dst_pd = mkldnn_matmul_prim_desc.dst_desc(); in AllocateOutputTensor() local399 output_mkl_shape.SetMklLayout(&dst_pd); in AllocateOutputTensor()405 output_tf_shape.AddDim((dst_pd.get_size() / sizeof(Toutput))); in AllocateOutputTensor()
719 auto dst_pd = concat_pd.dst_desc(); in Compute() local725 dnn_shape_dst.SetMklLayout(&dst_pd); in Compute()729 tf_shape_dst.AddDim((dst_pd.get_size() / sizeof(T))); in Compute()
500 auto dst_pd = eltwise_fwd_pd->dst_desc(); in Compute() local501 dnn_shape_dst.SetMklLayout(&dst_pd); in Compute()512 tf_shape_dst.AddDim(dst_pd.get_size() / sizeof(T)); in Compute()
849 auto dst_pd = bn_fwd->GetDstPd(); in Compute() local850 dnn_shape_dst.SetMklLayout(&dst_pd); in Compute()855 tf_shape_dst.AddDim(dst_pd.get_size() / sizeof(T)); in Compute()