Searched defs:SetReorder (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ | ||
| H A D | mkl_slice_op.cc | 313 void SetReorder(const MklSliceParams& sliceParams, MklPrimitive* op) { in SetReorder() function in tensorflow::MklSlicePrimitiveFactory |
| H A D | mkl_quantize_op.cc | 213 void SetReorder(const memory* from, const memory* to, MklPrimitive* op, in SetReorder() function in tensorflow::MklReorderWithScalePrimitiveFactory |