CUDNN API
8
|
#include <cudnn_frontend_Operation.h>
Public Member Functions | |
auto | setxDesc (ManagedOpaqueDescriptor const &raw_tensor) -> OperationBuilder_v8 & |
auto | setxDesc (Tensor_v8 const &tensor) -> OperationBuilder_v8 & |
auto | setbDesc (Tensor_v8 const &tensor) -> OperationBuilder_v8 & |
auto | setyDesc (Tensor_v8 const &tensor) -> OperationBuilder_v8 & |
auto | setwDesc (Tensor_v8 const &tensor) -> OperationBuilder_v8 & |
auto | setcDesc (ConvDesc_v8 const &conv) -> OperationBuilder_v8 & |
auto | setpwDesc (PointWiseDesc_v8 const &pointWiseDesc) -> OperationBuilder_v8 & |
auto | setAlpha (float alpha) -> OperationBuilder_v8 & |
auto | setAlpha (double alpha) -> OperationBuilder_v8 & |
auto | setAlpha2 (float alpha) -> OperationBuilder_v8 & |
auto | setAlpha2 (double alpha) -> OperationBuilder_v8 & |
auto | setBeta (float beta) -> OperationBuilder_v8 & |
auto | setBeta (double beta) -> OperationBuilder_v8 & |
OperationBuilder_v8 (cudnnBackendDescriptorType_t mode) | |
Operation_v8 && | build () |
Private Attributes | |
Operation_v8 | m_operation |
bool | is_convolution_op = false |
OperationBuilder_v8 Class Helper class used to build Operation_v8 class
Definition at line 133 of file cudnn_frontend_Operation.h.
|
inline |
constructs the backend Operation_v8 by calling the cudnn API Throws the appropriate error message
Definition at line 259 of file cudnn_frontend_Operation.h.
References cudnn_frontend::Operation_v8::alpha2_d, cudnn_frontend::Operation_v8::alpha2_s, cudnn_frontend::Operation_v8::alpha_d, cudnn_frontend::Operation_v8::alpha_s, cudnn_frontend::Operation_v8::alphabetaType, cudnn_frontend::Operation_v8::bdesc, cudnn_frontend::Operation_v8::beta_d, cudnn_frontend::Operation_v8::beta_s, cudnn_frontend::Operation_v8::cdesc, cudnn_frontend::BackendDescriptor::initialize_managed_backend_pointer(), cudnn_frontend::Operation_v8::op_mode, cudnn_frontend::Operation_v8::operationTag, cudnn_frontend::BackendDescriptor::pointer, cudnn_frontend::Operation_v8::pointwise_mode, cudnn_frontend::Operation_v8::pointwise_port_count, cudnn_frontend::Operation_v8::pwdesc, cudnn_frontend::set_error_and_throw_exception(), cudnn_frontend::BackendDescriptor::status, cudnn_frontend::Operation_v8::wdesc, cudnn_frontend::Operation_v8::xdesc, and cudnn_frontend::Operation_v8::ydesc.
|
private |
Definition at line 136 of file cudnn_frontend_Operation.h.
|
private |
Definition at line 135 of file cudnn_frontend_Operation.h.