Searched defs:ReorderContext (Results 1 – 1 of 1) sorted by relevance
110 struct ReorderContext { struct in tensorflow::MklReorderWithScalePrimitive112 std::shared_ptr<dnnl::memory> src_mem;113 std::shared_ptr<dnnl::memory> dst_mem;116 std::shared_ptr<reorder::primitive_desc> reorder_pd;117 std::shared_ptr<primitive> reorder_prim;120 std::shared_ptr<dnnl::stream> reorder_stream;122 std::unordered_map<int, dnnl::memory> prim_args;124 ReorderContext() in ReorderContext() argument