Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
caffe2::miopenTensorDescWrapper Class Reference

miopenTensorDescWrapper is the placeholder that wraps around a miopenTensorDescriptor_t, allowing us to do descriptor change as-needed during runtime. More...

#include <common_miopen.h>

Public Member Functions

miopenTensorDescriptor_t Descriptor (const miopenDataType_t type, const vector< int > &dims, bool *changed)
 
template<typename T >
miopenTensorDescriptor_t Descriptor (const StorageOrder &order, const vector< int > &dims)
 

Detailed Description

miopenTensorDescWrapper is the placeholder that wraps around a miopenTensorDescriptor_t, allowing us to do descriptor change as-needed during runtime.

Definition at line 135 of file common_miopen.h.


The documentation for this class was generated from the following file: