Searched refs:bn_bwd (Results 1 – 1 of 1) sorted by relevance
404 if (context_.bn_bwd == nullptr) Setup(bwdParams); in MklFusedBatchNormBwdPrimitive()504 std::shared_ptr<mkldnn::primitive> bn_bwd; member568 context_.bn_bwd.reset(new batch_normalization_backward(*context_.bwd_pd)); in Setup()577 context_.bwd_primitives.push_back(*context_.bn_bwd); in Setup()588 auto bn_bwd = static_cast<MklFusedBatchNormBwdPrimitive<T, U>*>( in Get() local591 if (bn_bwd == nullptr) { in Get()592 bn_bwd = new MklFusedBatchNormBwdPrimitive<T, U>(bwdParams); in Get()594 bwdParams, bn_bwd); in Get()596 return bn_bwd; in Get()1222 MklFusedBatchNormBwdPrimitive<T, U>* bn_bwd = in Compute() local[all …]