Searched refs:reorder_prim (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_quantize_op.cc | 86 std::shared_ptr<primitive> GetPrimitive() { return context_.reorder_prim; } in GetPrimitive()
97 context_.reorder_prim->execute(*reorder_stream, context_.prim_args); in Execute()
112 std::shared_ptr<primitive> reorder_prim; member
123 reorder_prim(nullptr) {} in ReorderContext()
149 context_.reorder_prim.reset(new reorder(*context_.reorder_pd)); in Setup()
473 MklReorderWithScalePrimitive* reorder_prim = in Compute() local
477 cpu_stream.reset(CreateStream(ctx, reorder_prim->GetEngine())); in Compute()
478 reorder_prim->Execute(src.GetUsrMemDataHandle(), dst.GetUsrMemDataHandle(), in Compute()
|
| D | mkl_slice_op.cc | 210 std::shared_ptr<primitive> GetPrimitive() { return context_.reorder_prim; } in GetPrimitive()
216 std::shared_ptr<primitive> reorder_prim; member
223 : src_mem(nullptr), dst_mem(nullptr), reorder_prim(nullptr) {} in SliceContext()
239 context_.reorder_prim = in Setup()
245 context_.slice_primitives.push_back(*context_.reorder_prim); in Setup()
452 MklSlicePrimitive<T>* reorder_prim = in ComputeMklSlice() local
456 slice_stream.reset(CreateStream(context, reorder_prim->GetEngine())); in ComputeMklSlice()
457 reorder_prim->Execute(sliceParams, slice_stream); in ComputeMklSlice()
|
| D | mkl_requantize_per_channel_op.cc | 136 std::unique_ptr<mkldnn::primitive> reorder_prim( in Compute() local
138 reorder_prim->execute(*reorder_stream, reorder_args); in Compute()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1871 std::shared_ptr<primitive> GetPrimitive() { return context_.reorder_prim; } in GetPrimitive()
1884 std::shared_ptr<primitive> reorder_prim; member
1886 : src_mem(nullptr), dst_mem(nullptr), reorder_prim(nullptr) {} in ReorderContext()
1895 context_.reorder_prim = std::make_shared<mkldnn::reorder>( in Setup()
1993 MklReorderPrimitive* reorder_prim = in FindOrCreateReorder() local
1995 return reorder_prim; in FindOrCreateReorder()
|