Searched refs:GetDiffSrcPd (Results 1 – 1 of 1) sorted by relevance
484 memory::desc GetDiffSrcPd() { return context_.diff_src_mem->get_desc(); } in GetDiffSrcPd() function in tensorflow::MklFusedBatchNormBwdPrimitive1247 auto diff_src_pd = bn_bwd->GetDiffSrcPd(); in Compute()