Searched refs:conv_fwd (Results 1 – 1 of 1) sorted by relevance
108 if (context_.conv_fwd == nullptr) { in MklConvFwdPrimitive()193 std::shared_ptr<mkldnn::primitive> conv_fwd; member210 conv_fwd(nullptr), in ConvFwdContext()302 context_.conv_fwd.reset(new convolution_forward( in Setup()306 context_.conv_fwd.reset( in Setup()311 context_.fwd_primitives.push_back(*context_.conv_fwd); in Setup()325 MklConvFwdPrimitive<T, Tinput, Tfilter, Tbias, Toutput>* conv_fwd = nullptr; in Get() local329 conv_fwd = new MklConvFwdPrimitive<T, Tinput, Tfilter, Tbias, Toutput>( in Get()333 conv_fwd = dynamic_cast< in Get()338 if (conv_fwd == nullptr) { in Get()[all …]