CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
Classes | Namespaces
linear_combination_relu.h File Reference

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"
Include dependency graph for linear_combination_relu.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