Searched refs:MklSliceParams (Results 1 – 1 of 1) sorted by relevance
160 struct MklSliceParams { struct170 MklSliceParams(const memory* from, const memory* to, memory::dims begin_dims, in MklSliceParams() function179 explicit MklSlicePrimitive(const MklSliceParams& sliceParams) in MklSlicePrimitive()186 void Execute(const MklSliceParams& sliceParams, in Execute()226 void Setup(const MklSliceParams& sliceParams) { in Setup()252 static MklSlicePrimitive<T>* Get(const MklSliceParams& sliceParams) { in Get()272 static string CreateKey(const MklSliceParams& sliceParams) { in CreateKey()300 MklPrimitive* GetReorder(const MklSliceParams& sliceParams) { in GetReorder()305 void SetReorder(const MklSliceParams& sliceParams, MklPrimitive* op) { in SetReorder()450 MklSliceParams sliceParams(&src.GetOpMem(), output.GetUsrMem(), in ComputeMklSlice()