![]() |
CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
|
Host-constructable parameters structure.
#include <linear_combination_relu.h>
Public Member Functions | |
| CUTLASS_HOST_DEVICE | Params () |
| CUTLASS_HOST_DEVICE | Params (ElementCompute alpha, ElementCompute beta, ElementCompute threshold=ElementCompute(0)) |
| CUTLASS_HOST_DEVICE | Params (ElementCompute const *alpha_ptr, ElementCompute const *beta_ptr, ElementCompute threshold=ElementCompute(0)) |
Public Attributes | |
| ElementCompute | alpha |
| scales accumulators More... | |
| ElementCompute | beta |
| scales source tensor More... | |
| ElementCompute | threshold |
| Relu threshold. More... | |
| ElementCompute const * | alpha_ptr |
| pointer to accumulator scalar - if not null, loads it from memory More... | |
| ElementCompute const * | beta_ptr |
| pointer to source scalar - if not null, loads it from memory More... | |
|
inline |
|
inline |
|
inline |
| ElementCompute cutlass::epilogue::thread::LinearCombinationRelu< ElementOutput_, Count, int, float, Round >::Params::alpha |
| ElementCompute const* cutlass::epilogue::thread::LinearCombinationRelu< ElementOutput_, Count, int, float, Round >::Params::alpha_ptr |
| ElementCompute cutlass::epilogue::thread::LinearCombinationRelu< ElementOutput_, Count, int, float, Round >::Params::beta |
| ElementCompute const* cutlass::epilogue::thread::LinearCombinationRelu< ElementOutput_, Count, int, float, Round >::Params::beta_ptr |
| ElementCompute cutlass::epilogue::thread::LinearCombinationRelu< ElementOutput_, Count, int, float, Round >::Params::threshold |
1.8.11