Method SetUsrMem

tensorflow/core/util/mkl_util.h:1304–1308

Set user memory primitive using specified dimensions, memory format tag and data_buffer. Function automatically uses element data type by using input type T used for creating call object. In a nutshell, function allows user to describe the input tensor to an operation. E.g., filter of Conv2D is of shape {1, 2, 3, 4}, and memory format tag HWIO, and the buffer that contains actual values is pointe

  /// memory format tag HWIO, and the buffer that contains actual values is
  /// pointed by data_buffer.
  inline void SetUsrMem(const memory::dims& dim, memory::format_tag fm,
                        void* data_buffer = nullptr) {
    auto md = memory::desc(dim, MklDnnType<T>(), fm);
    SetUsrMem(md, data_buffer);
  }

  inline void SetUsrMem(const memory::dims& dim, memory::format_tag fm,
                        const Tensor* tensor) {