Searched refs:diff_src_pd (Results 1 – 3 of 3) sorted by relevance
438 auto diff_src_pd = bwd_input_pd.get()->diff_src_desc(); in Compute() local445 diff_src_mkl_shape.SetMklLayout(&diff_src_pd); in Compute()450 diff_src_tf_shape.AddDim(diff_src_pd.get_size() / sizeof(T)); in Compute()
710 auto diff_src_pd = eltwise_bwd_pd->diff_src_desc(); in Compute() local712 dnn_shape_diff_src.SetMklLayout(&diff_src_pd); in Compute()724 tf_shape_diff_src.AddDim(diff_src_pd.get_size() / sizeof(T)); in Compute()
1247 auto diff_src_pd = bn_bwd->GetDiffSrcPd(); in Compute() local1248 dnn_shape_diff_src.SetMklLayout(&diff_src_pd); in Compute()1252 tf_shape_diff_src.AddDim(diff_src_pd.get_size() / sizeof(T)); in Compute()