Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmkl_transpose_op.cc144 net.push_back(FindOrCreateReorder<T>(in.GetUsrMem(), out.GetUsrMem())); in MKLTransposeND()
Dmkl_slice_op.cc426 MklSliceParams sliceParams(src.GetUsrMem(), output.GetUsrMem(), in ComputeMklSlice()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h1785 inline const memory* GetUsrMem() const { return user_memory_; } in GetUsrMem() function