Method GetUsrMem

tensorflow/core/util/mkl_util.h:1373–1373

Get function for user memory primitive.

  /// Get function for user memory primitive.
  inline const memory* GetUsrMem() const { return user_memory_; }

  /// Get function for descriptor of user memory.
  inline memory::desc GetUsrMemDesc() const {