Searched refs:GetBatchNormFwdPd (Results 1 – 1 of 1) sorted by relevance
143 std::shared_ptr<BatchNormFwdPd> GetBatchNormFwdPd() const { in GetBatchNormFwdPd() function in tensorflow::MklFusedBatchNormFwdPrimitive787 bn_fwd->GetBatchNormFwdPd()->workspace_desc(); in Compute()836 std::shared_ptr<BatchNormFwdPd> bn_fwd_pd = bn_fwd->GetBatchNormFwdPd(); in Compute()