Searched refs:MklFusedBatchNormBwdPrimitive (Results 1 – 1 of 1) sorted by relevance
400 class MklFusedBatchNormBwdPrimitive : public MklPrimitive { class402 explicit MklFusedBatchNormBwdPrimitive(const MklBatchNormBwdParams& bwdParams) in MklFusedBatchNormBwdPrimitive() function in tensorflow::MklFusedBatchNormBwdPrimitive407 ~MklFusedBatchNormBwdPrimitive() {} in ~MklFusedBatchNormBwdPrimitive()586 static MklFusedBatchNormBwdPrimitive<T, U>* Get( in Get()588 auto bn_bwd = static_cast<MklFusedBatchNormBwdPrimitive<T, U>*>( in Get()592 bn_bwd = new MklFusedBatchNormBwdPrimitive<T, U>(bwdParams); in Get()1222 MklFusedBatchNormBwdPrimitive<T, U>* bn_bwd = in Compute()