Searched refs:bn_bwd_pd (Results 1 – 1 of 1) sorted by relevance
1226 std::shared_ptr<BatchNormBwdPd> bn_bwd_pd = bn_bwd->GetBatchNormBwdPd(); in Compute() local1227 if (!native_format && diff_dst_md != bn_bwd_pd->diff_dst_desc()) { in Compute()1229 diff_dst.CheckReorderToOpMem(bn_bwd_pd->diff_dst_desc(), cpu_engine_, in Compute()1234 if (!native_format && (src_md != bn_bwd_pd->src_desc())) { in Compute()1236 src.CheckReorderToOpMem(bn_bwd_pd->src_desc(), cpu_engine_, context); in Compute()