Home
last modified time | relevance | path

Searched defs:dst_pd (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_lrn_op.cc196 memory::desc dst_pd = lrn_fwd_prim_desc.dst_desc(); in AllocateOutputTensor() local
459 memory::desc dst_pd = lrn_bkwd_prim_desc.diff_src_desc(); in AllocateOutputTensor() local
Dmkl_softmax_op.cc303 auto dst_pd = softmax_fwd->GetSoftmaxFwdPd()->dst_desc(); in Compute() local
Dmkl_pooling_ops_common.h664 memory::desc dst_pd = pool_fwd_prim_desc.dst_desc(); in AllocateOutputTensor() local
717 memory::desc dst_pd = pool_bkwd_prim_desc.diff_src_desc(); in AllocateOutputTensor() local
Dmkl_conv_grad_input_ops.cc570 auto dst_pd = conv_pd.diff_src_desc(); in AllocateOutputTensor() local
Dmkl_concat_op.cc731 auto dst_pd = concat_pd.dst_desc(); in Compute() local
Dmkl_relu_op.cc502 auto dst_pd = eltwise_fwd_pd->dst_desc(); in Compute() local
Dmkl_fused_batch_norm_op.cc910 auto dst_pd = bn_fwd->GetDstPd(); in Compute() local