CUDNN Frontend API
8.3.0
|
#include <cudnn_frontend_Tensor.h>
Public Member Functions | |
auto | setDataType (cudnnDataType_t data_type_) -> TensorBuilder_v8 & |
Set Datatype for the Tensor_v8. More... | |
auto | setDim (int64_t ndim, int64_t const *dim) -> TensorBuilder_v8 & |
Set Dimensions of the tensor. More... | |
auto | setStrides (int64_t ndim, int64_t const *strides) -> TensorBuilder_v8 & |
Set Strides of the tensor. More... | |
auto | setId (int64_t id_) -> TensorBuilder_v8 & |
Set Unique Id of the tensor. More... | |
auto | setAlignment (int64_t alignment_) -> TensorBuilder_v8 & |
Set Alignment of the tensor. More... | |
auto | setVirtual (bool virtual_=true) -> TensorBuilder_v8 & |
Set Alignment of the tensor. More... | |
auto | setByValue (bool isByValue_=true) -> TensorBuilder_v8 & |
auto | setVectorCountAndDimension (int64_t vectorCount_, int64_t vectorDimension_) -> TensorBuilder_v8 & |
Tensor_v8 && | build () |
TensorBuilder_v8 ()=default | |
~TensorBuilder_v8 ()=default | |
TensorBuilder_v8 (TensorBuilder_v8 &&)=delete | |
TensorBuilder_v8 (TensorBuilder_v8 const &)=delete | |
TensorBuilder_v8 & | operator= (TensorBuilder_v8 const &)=delete |
Private Attributes | |
Tensor_v8 | m_tensor |
TensorBuilder_v8 Class Helper class used to build Tensor_v8 class
Definition at line 140 of file cudnn_frontend_Tensor.h.
|
explicitdefault |
|
default |
|
delete |
|
delete |
|
inline |
constructs the Tensor_v8 by calling the cudnn API Throws the appropriate error message
Definition at line 207 of file cudnn_frontend_Tensor.h.
References cudnn_frontend::getLogger(), cudnn_frontend::Tensor_v8::operator=(), cudnn_frontend::set_error_and_throw_exception(), cudnn_frontend::BackendDescriptor::status, and cudnn_frontend::Tensor_v8::TensorBuilder_v8.
|
delete |
|
private |
Definition at line 396 of file cudnn_frontend_Tensor.h.