CUDNN Frontend API  8.2.0
cudnn_frontend::OperationBuilder_v8 Class Reference

#include <cudnn_frontend_Operation.h>

Collaboration diagram for cudnn_frontend::OperationBuilder_v8:
Collaboration graph

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 setdyDesc (ManagedOpaqueDescriptor const &raw_tensor) -> OperationBuilder_v8 &
 
auto setdyDesc (Tensor_v8 const &tensor) -> OperationBuilder_v8 &
 
auto setdxDesc (Tensor_v8 const &tensor) -> OperationBuilder_v8 &
 
auto setdwDesc (Tensor_v8 const &tensor) -> OperationBuilder_v8 &
 
auto setcDesc (ConvDesc_v8 const &conv) -> OperationBuilder_v8 &
 
auto setaMatDesc (Tensor_v8 const &tensor) -> OperationBuilder_v8 &
 
auto setbMatDesc (Tensor_v8 const &tensor) -> OperationBuilder_v8 &
 
auto setcMatDesc (Tensor_v8 const &tensor) -> OperationBuilder_v8 &
 
auto setmatmulDesc (MatMulDesc_v8 const &matmulDesc) -> OperationBuilder_v8 &
 
auto setreductionDesc (ReductionDesc_v8 const &reductionDesc) -> 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
 
bool is_pointwise_op = false
 
bool is_matmul_op = false
 
bool is_reduction_op = false
 

Detailed Description

OperationBuilder_v8 Class Helper class used to build Operation_v8 class

Definition at line 173 of file cudnn_frontend_Operation.h.

Member Function Documentation

◆ build()

Operation_v8&& cudnn_frontend::OperationBuilder_v8::build ( )
inline

constructs the backend Operation_v8 by calling the cudnn API Throws the appropriate error message

Definition at line 407 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::amatdesc, cudnn_frontend::Operation_v8::bdesc, cudnn_frontend::Operation_v8::beta_d, cudnn_frontend::Operation_v8::beta_s, cudnn_frontend::Operation_v8::bmatdesc, cudnn_frontend::Operation_v8::cdesc, cudnn_frontend::Operation_v8::cmatdesc, cudnn_frontend::Operation_v8::dwdesc, cudnn_frontend::Operation_v8::dxdesc, cudnn_frontend::Operation_v8::dydesc, cudnn_frontend::BackendDescriptor::initialize_managed_backend_pointer(), cudnn_frontend::Operation_v8::is_pointwise_activation_bwd_op, cudnn_frontend::Operation_v8::is_pointwise_activation_fwd_op, cudnn_frontend::Operation_v8::is_pointwise_math_op, cudnn_frontend::Operation_v8::matmuldesc, 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::Operation_v8::reductiondesc, 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.

Member Data Documentation

◆ is_convolution_op

bool cudnn_frontend::OperationBuilder_v8::is_convolution_op = false
private

Definition at line 176 of file cudnn_frontend_Operation.h.

◆ is_matmul_op

bool cudnn_frontend::OperationBuilder_v8::is_matmul_op = false
private

Definition at line 178 of file cudnn_frontend_Operation.h.

◆ is_pointwise_op

bool cudnn_frontend::OperationBuilder_v8::is_pointwise_op = false
private

Definition at line 177 of file cudnn_frontend_Operation.h.

◆ is_reduction_op

bool cudnn_frontend::OperationBuilder_v8::is_reduction_op = false
private

Definition at line 179 of file cudnn_frontend_Operation.h.

◆ m_operation

Operation_v8 cudnn_frontend::OperationBuilder_v8::m_operation
private

Definition at line 175 of file cudnn_frontend_Operation.h.


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