CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
|
Functor performing linear combination operations used by epilogues. Values are clamped before converting to the output element type. More...
#include "cutlass/cutlass.h"
#include "cutlass/numeric_types.h"
#include "cutlass/array.h"
#include "cutlass/functional.h"
#include "cutlass/numeric_conversion.h"
Go to the source code of this file.
Classes | |
class | cutlass::epilogue::thread::LinearCombinationRelu< ElementOutput_, Count, ElementAccumulator_, ElementCompute_, Round > |
struct | cutlass::epilogue::thread::LinearCombinationRelu< ElementOutput_, Count, ElementAccumulator_, ElementCompute_, Round >::Params |
Host-constructable parameters structure. More... | |
class | cutlass::epilogue::thread::LinearCombinationRelu< ElementOutput_, Count, int, float, Round > |
struct | cutlass::epilogue::thread::LinearCombinationRelu< ElementOutput_, Count, int, float, Round >::Params |
Host-constructable parameters structure. More... | |
Namespaces | |
cutlass | |
cutlass::epilogue | |
cutlass::epilogue::thread | |