Searched refs:reorder_prim (Results 1 – 2 of 2) sorted by relevance
202 std::shared_ptr<primitive> GetPrimitive() { return context_.reorder_prim; } in GetPrimitive()208 std::shared_ptr<primitive> reorder_prim; member214 : src_mem(nullptr), dst_mem(nullptr), reorder_prim(nullptr) {} in SliceContext()235 context_.reorder_prim = std::make_shared<mkldnn::reorder>( in Setup()237 context_.slice_primitives.push_back(*context_.reorder_prim); in Setup()428 MklSlicePrimitive<T>* reorder_prim = in ComputeMklSlice() local431 reorder_prim->Execute(sliceParams); in ComputeMklSlice()
2265 std::shared_ptr<primitive> GetPrimitive() { return context_.reorder_prim; } in GetPrimitive()2276 std::shared_ptr<primitive> reorder_prim; member2278 : src_mem(nullptr), dst_mem(nullptr), reorder_prim(nullptr) {} in ReorderContext()2288 context_.reorder_prim = std::make_shared<mkldnn::reorder>( in Setup()2363 MklReorderPrimitive* reorder_prim = in FindOrCreateReorder() local2365 return *reorder_prim->GetPrimitive(); in FindOrCreateReorder()