Home
last modified time | relevance | path

Searched refs:SetReorder (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_quantize_op.cc167 MklReorderWithScalePrimitiveFactory<T>::GetInstance().SetReorder( in Get()
204 void SetReorder(const memory* from, const memory* to, MklPrimitive* op, in SetReorder() function in tensorflow::MklReorderWithScalePrimitiveFactory
Dmkl_slice_op.cc257 MklSlicePrimitiveFactory<T>::GetInstance().SetReorder(sliceParams, in Get()
305 void SetReorder(const MklSliceParams& sliceParams, MklPrimitive* op) { in SetReorder() function in tensorflow::MklSlicePrimitiveFactory
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h1909 MklReorderPrimitiveFactory<T>::GetInstance().SetReorder(from, to, in Get()
1978 void SetReorder(const memory* from, const memory* to, MklPrimitive* op) { in SetReorder() function