Searched refs:bn_fwd (Results 1 – 1 of 1) sorted by relevance
67 if (context_.bn_fwd == nullptr) Setup(fwdParams); in MklFusedBatchNormFwdPrimitive()168 std::shared_ptr<mkldnn::primitive> bn_fwd; member182 bn_fwd(nullptr) {} in BatchNormFwdContext()246 context_.bn_fwd.reset(new batch_normalization_forward(*context_.fwd_pd)); in Setup()281 context_.bn_fwd.reset(new batch_normalization_forward(*context_.fwd_pd)); in Setup()316 context_.bn_fwd.reset(new batch_normalization_forward(*context_.fwd_pd)); in Setup()319 context_.fwd_primitives.push_back(*context_.bn_fwd); in Setup()330 auto bn_fwd = static_cast<MklFusedBatchNormFwdPrimitive<T, U>*>( in Get() local334 if (bn_fwd == nullptr) { in Get()335 bn_fwd = new MklFusedBatchNormFwdPrimitive<T, U>(fwdParams); in Get()[all …]