CUDNN API  8
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 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
 

Detailed Description

OperationBuilder_v8 Class Helper class used to build Operation_v8 class

Definition at line 133 of file cudnn_frontend_Operation.h.

Member Function Documentation

◆ build()

Member Data Documentation

◆ is_convolution_op

bool cudnn_frontend::OperationBuilder_v8::is_convolution_op = false
private

Definition at line 136 of file cudnn_frontend_Operation.h.

◆ m_operation

Operation_v8 cudnn_frontend::OperationBuilder_v8::m_operation
private

Definition at line 135 of file cudnn_frontend_Operation.h.


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