CUDNN API  8
OperationBuilder_v8

Functions

auto cudnn_frontend::OperationBuilder_v8::setxDesc (ManagedOpaqueDescriptor const &raw_tensor) -> OperationBuilder_v8 &
 
auto cudnn_frontend::OperationBuilder_v8::setxDesc (Tensor_v8 const &tensor) -> OperationBuilder_v8 &
 
auto cudnn_frontend::OperationBuilder_v8::setbDesc (Tensor_v8 const &tensor) -> OperationBuilder_v8 &
 
auto cudnn_frontend::OperationBuilder_v8::setyDesc (Tensor_v8 const &tensor) -> OperationBuilder_v8 &
 
auto cudnn_frontend::OperationBuilder_v8::setwDesc (Tensor_v8 const &tensor) -> OperationBuilder_v8 &
 
auto cudnn_frontend::OperationBuilder_v8::setcDesc (ConvDesc_v8 const &conv) -> OperationBuilder_v8 &
 
auto cudnn_frontend::OperationBuilder_v8::setpwDesc (PointWiseDesc_v8 const &pointWiseDesc) -> OperationBuilder_v8 &
 
auto cudnn_frontend::OperationBuilder_v8::setAlpha (float alpha) -> OperationBuilder_v8 &
 
auto cudnn_frontend::OperationBuilder_v8::setAlpha (double alpha) -> OperationBuilder_v8 &
 
auto cudnn_frontend::OperationBuilder_v8::setAlpha2 (float alpha) -> OperationBuilder_v8 &
 
auto cudnn_frontend::OperationBuilder_v8::setAlpha2 (double alpha) -> OperationBuilder_v8 &
 
auto cudnn_frontend::OperationBuilder_v8::setBeta (float beta) -> OperationBuilder_v8 &
 
auto cudnn_frontend::OperationBuilder_v8::setBeta (double beta) -> OperationBuilder_v8 &
 
 cudnn_frontend::OperationBuilder_v8::OperationBuilder_v8 (cudnnBackendDescriptorType_t mode)
 

Detailed Description

Set individual property of Operation_v8 class

Function Documentation

◆ OperationBuilder_v8()

cudnn_frontend::OperationBuilder_v8::OperationBuilder_v8 ( cudnnBackendDescriptorType_t  mode)
inline

Definition at line 248 of file cudnn_frontend_Operation.h.

References cudnn_frontend::Operation_v8::op_mode.

◆ setAlpha() [1/2]

auto cudnn_frontend::OperationBuilder_v8::setAlpha ( float  alpha) -> OperationBuilder_v8 &
inline

◆ setAlpha() [2/2]

auto cudnn_frontend::OperationBuilder_v8::setAlpha ( double  alpha) -> OperationBuilder_v8 &
inline

◆ setAlpha2() [1/2]

auto cudnn_frontend::OperationBuilder_v8::setAlpha2 ( float  alpha) -> OperationBuilder_v8 &
inline

◆ setAlpha2() [2/2]

auto cudnn_frontend::OperationBuilder_v8::setAlpha2 ( double  alpha) -> OperationBuilder_v8 &
inline

◆ setbDesc()

auto cudnn_frontend::OperationBuilder_v8::setbDesc ( Tensor_v8 const &  tensor) -> OperationBuilder_v8 &
inline

◆ setBeta() [1/2]

auto cudnn_frontend::OperationBuilder_v8::setBeta ( float  beta) -> OperationBuilder_v8 &
inline

◆ setBeta() [2/2]

auto cudnn_frontend::OperationBuilder_v8::setBeta ( double  beta) -> OperationBuilder_v8 &
inline

◆ setcDesc()

auto cudnn_frontend::OperationBuilder_v8::setcDesc ( ConvDesc_v8 const &  conv) -> OperationBuilder_v8 &
inline

◆ setpwDesc()

◆ setwDesc()

auto cudnn_frontend::OperationBuilder_v8::setwDesc ( Tensor_v8 const &  tensor) -> OperationBuilder_v8 &
inline

◆ setxDesc() [1/2]

auto cudnn_frontend::OperationBuilder_v8::setxDesc ( ManagedOpaqueDescriptor const &  raw_tensor) -> OperationBuilder_v8 &
inline

Definition at line 144 of file cudnn_frontend_Operation.h.

References cudnn_frontend::Operation_v8::xdesc.

◆ setxDesc() [2/2]

auto cudnn_frontend::OperationBuilder_v8::setxDesc ( Tensor_v8 const &  tensor) -> OperationBuilder_v8 &
inline

Definition at line 150 of file cudnn_frontend_Operation.h.

References cudnn_frontend::Operation_v8::xdesc.

◆ setyDesc()

auto cudnn_frontend::OperationBuilder_v8::setyDesc ( Tensor_v8 const &  tensor) -> OperationBuilder_v8 &
inline

Definition at line 166 of file cudnn_frontend_Operation.h.

References cudnn_frontend::Operation_v8::ydesc.