Searched refs:SetBatchNormFwd (Results 1 – 1 of 1) sorted by relevance
336 MklFusedBatchNormFwdPrimitiveFactory<T, U>::GetInstance().SetBatchNormFwd( in Get()370 void SetBatchNormFwd(const MklBatchNormFwdParams& fwdParams, in SetBatchNormFwd() function in tensorflow::MklFusedBatchNormFwdPrimitiveFactory