Back to Cutlass

CUTLASS: linear_combination_relu.h File Reference

docs/linear__combination__relu_8h.html

4.4.22.6 KB
Original Source

| | 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.](linear combination relu_8h_source.html)

|

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 | | |


Generated by 1.8.11