Searched defs:ConvFwdContext (Results 1 – 1 of 1) sorted by relevance
171 struct ConvFwdContext { struct in tensorflow::MklConvFwdPrimitive173 memory::format src_fmt;174 memory::format filter_fmt;177 std::shared_ptr<mkldnn::memory> src_mem;178 std::shared_ptr<mkldnn::memory> filter_mem;179 std::shared_ptr<mkldnn::memory> bias_mem;180 std::shared_ptr<mkldnn::memory> dst_mem;183 std::shared_ptr<mkldnn::convolution_forward::desc> fwd_desc;186 std::shared_ptr<mkldnn::memory::desc> src_md;187 std::shared_ptr<mkldnn::memory::desc> filter_md;[all …]