CUDNN Frontend API
8.2.0
|
#include <cudnn_frontend_Tensor.h>
Public Member Functions | |
std::string | describe () const override |
Return a string describing the backend Descriptor. More... | |
Tensor_v8 (Tensor_v8 &&from) | |
~Tensor_v8 ()=default | |
![]() | |
cudnnBackendDescriptor_t | get_raw_desc () const |
cudnnStatus_t | get_status () const |
Current status of the descriptor. More... | |
void | set_status (cudnnStatus_t const status_) const |
Set status of the descriptor. More... | |
void | set_error (const char *message) const |
Set Diagonistic error message. More... | |
const char * | get_error () const |
Diagonistic error message if any. More... | |
ManagedOpaqueDescriptor | get_desc () const |
Returns a copy of underlying managed descriptor. More... | |
cudnnStatus_t | initialize_managed_backend_pointer (cudnnBackendDescriptorType_t type) |
Initializes the underlying managed descriptor. More... | |
Private Member Functions | |
Tensor_v8 ()=default | |
Tensor_v8 (Tensor_v8 const &)=delete | |
Tensor_v8 & | operator= (Tensor_v8 const &)=delete |
Private Attributes | |
cudnnDataType_t | data_type = CUDNN_DATA_FLOAT |
int64_t | btensor_dimA [CUDNN_DIM_MAX+1] = {-1} |
Datatype of the elements. More... | |
int64_t | btensor_strA [CUDNN_DIM_MAX+1] = {-1} |
n, g, c, d, h, w More... | |
int64_t | id = -1 |
n, g, c, d, h, w More... | |
int64_t | alignment = -1 |
Unique id of the tensor. More... | |
int64_t | nDims = -1 |
Certain engine config expect minimum alignment of 16B. More... | |
bool | isVirtual = false |
Number of Dimensions of the tensor. More... | |
Friends | |
class | TensorBuilder_v8 |
Additional Inherited Members | |
![]() | |
BackendDescriptor (ManagedOpaqueDescriptor pointer_, cudnnStatus_t status_, std::string err_msg_) | |
BackendDescriptor ()=default | |
![]() | |
ManagedOpaqueDescriptor | pointer |
cudnnStatus_t | status = CUDNN_STATUS_SUCCESS |
Shared pointer of the OpaqueBackendPointer. More... | |
std::string | err_msg |
Error message if any being set. More... | |
Tensor_v8 Class This class tells the properties of the Tensor_v8 on which the operation will be performed Properties:
Use TensorBuilder_v8 to build this class. Describe returns a string describing the tensor class
Definition at line 51 of file cudnn_frontend_Tensor.h.
|
inline |
Definition at line 75 of file cudnn_frontend_Tensor.h.
References btensor_dimA, btensor_strA, operator=(), Tensor_v8(), and ~Tensor_v8().
|
default |
Referenced by Tensor_v8().
|
privatedefault |
Referenced by Tensor_v8().
|
privatedelete |
|
inlineoverridevirtual |
Return a string describing the backend Descriptor.
Implements cudnn_frontend::BackendDescriptor.
Definition at line 55 of file cudnn_frontend_Tensor.h.
References alignment, btensor_dimA, btensor_strA, data_type, nDims, and nlohmann::to_string().
Referenced by cudnn_frontend::TensorBuilder_v8::build(), and Tensor_v8().
|
friend |
Definition at line 53 of file cudnn_frontend_Tensor.h.
Referenced by cudnn_frontend::TensorBuilder_v8::build().
|
private |
Unique id of the tensor.
Definition at line 98 of file cudnn_frontend_Tensor.h.
Referenced by describe().
|
private |
Datatype of the elements.
Definition at line 95 of file cudnn_frontend_Tensor.h.
Referenced by describe(), and Tensor_v8().
|
private |
n, g, c, d, h, w
Definition at line 96 of file cudnn_frontend_Tensor.h.
Referenced by describe(), and Tensor_v8().
|
private |
Definition at line 94 of file cudnn_frontend_Tensor.h.
Referenced by describe().
|
private |
n, g, c, d, h, w
Definition at line 97 of file cudnn_frontend_Tensor.h.
|
private |
Number of Dimensions of the tensor.
Definition at line 101 of file cudnn_frontend_Tensor.h.
|
private |
Certain engine config expect minimum alignment of 16B.
Alignment of the tensor.
Definition at line 100 of file cudnn_frontend_Tensor.h.
Referenced by describe().