Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmkl_slice_op.cc162 struct MklSliceParams { struct
172 MklSliceParams(const memory* from, const memory* to, memory::dims begin_dims, in MklSliceParams() argument
181 explicit MklSlicePrimitive(const MklSliceParams& sliceParams) { in MklSlicePrimitive()
188 void Execute(const MklSliceParams& sliceParams) { in Execute()
219 void Setup(const MklSliceParams& sliceParams) { in Setup()
244 static MklSlicePrimitive<T>* Get(const MklSliceParams& sliceParams) { in Get()
264 static string CreateKey(const MklSliceParams& sliceParams) { in CreateKey()
293 MklPrimitive* GetReorder(const MklSliceParams& sliceParams) { in GetReorder()
298 void SetReorder(const MklSliceParams& sliceParams, MklPrimitive* op) { in SetReorder()
426 MklSliceParams sliceParams(src.GetUsrMem(), output.GetUsrMem(), in ComputeMklSlice()