Searched defs:ConvFwdContext (Results 1 – 1 of 1) sorted by relevance
173 struct ConvFwdContext { struct in tensorflow::MklConvFwdPrimitive175 std::shared_ptr<mkldnn::memory> src_mem;176 std::shared_ptr<mkldnn::memory> filter_mem;177 std::shared_ptr<mkldnn::memory> bias_mem;178 std::shared_ptr<mkldnn::memory> dst_mem;181 std::shared_ptr<mkldnn::convolution_forward::desc> fwd_desc;184 std::shared_ptr<mkldnn::memory::desc> src_md;185 std::shared_ptr<mkldnn::memory::desc> filter_md;186 std::shared_ptr<mkldnn::memory::desc> bias_md;187 std::shared_ptr<mkldnn::memory::desc> dst_md;[all …]