Searched defs:ConcatFwdContext (Results 1 – 1 of 1) sorted by relevance
324 struct ConcatFwdContext { struct in tensorflow::MklConcatFwdPrimitive326 std::vector<dnnl::memory> data_mem;327 std::vector<std::shared_ptr<dnnl::memory>> data_mem_shdptr;328 std::shared_ptr<dnnl::memory> dst_mem;331 std::vector<dnnl::memory::desc> src_md;332 std::shared_ptr<dnnl::memory::desc> dst_md;335 std::shared_ptr<dnnl::concat::primitive_desc> fwd_pd;336 std::shared_ptr<dnnl::primitive> concat_fwd;338 std::vector<dnnl::primitive> fwd_primitives;340 std::vector<std::unordered_map<int, memory>> fwd_primitives_args;[all …]