Searched refs:MklFusedBatchNormFwdPrimitive (Results 1 – 1 of 1) sorted by relevance
63 class MklFusedBatchNormFwdPrimitive : public MklPrimitive { class65 explicit MklFusedBatchNormFwdPrimitive(const MklBatchNormFwdParams& fwdParams) in MklFusedBatchNormFwdPrimitive() function in tensorflow::MklFusedBatchNormFwdPrimitive70 ~MklFusedBatchNormFwdPrimitive() {} in ~MklFusedBatchNormFwdPrimitive()328 static MklFusedBatchNormFwdPrimitive<T, U>* Get( in Get()330 auto bn_fwd = static_cast<MklFusedBatchNormFwdPrimitive<T, U>*>( in Get()335 bn_fwd = new MklFusedBatchNormFwdPrimitive<T, U>(fwdParams); in Get()780 MklFusedBatchNormFwdPrimitive<T, U>* bn_fwd = in Compute()