Searched refs:conv_fwd (Results 1 – 1 of 1) sorted by relevance
103 if (context_.conv_fwd == nullptr) { in MklConvFwdPrimitive()191 std::shared_ptr<mkldnn::primitive> conv_fwd; member206 conv_fwd(nullptr) {} in ConvFwdContext()298 context_.conv_fwd.reset(new convolution_forward(*context_.fwd_pd)); in Setup()305 context_.conv_fwd.reset(new convolution_forward(*context_.fwd_pd)); in Setup()311 context_.fwd_primitives.push_back(*context_.conv_fwd); in Setup()325 MklConvFwdPrimitive<Tinput, Tfilter, Tbias, Toutput>* conv_fwd = nullptr; in Get() local329 conv_fwd = in Get()333 conv_fwd = in Get()338 if (conv_fwd == nullptr) { in Get()[all …]