Searched refs:MklSlicePrimitive (Results 1 – 1 of 1) sorted by relevance
179 class MklSlicePrimitive : public MklPrimitive { class181 explicit MklSlicePrimitive(const MklSliceParams& sliceParams) { in MklSlicePrimitive() function in tensorflow::MklSlicePrimitive186 ~MklSlicePrimitive() {} in ~MklSlicePrimitive()244 static MklSlicePrimitive<T>* Get(const MklSliceParams& sliceParams) { in Get()245 auto reorderPrim = static_cast<MklSlicePrimitive<T>*>( in Get()248 reorderPrim = new MklSlicePrimitive<T>(sliceParams); in Get()428 MklSlicePrimitive<T>* reorder_prim = in ComputeMklSlice()