|
Poplar and PopLibs
|
Stores parameters for the reduce operation, as well as the basic operation being performed (for example, add or mul).
More...
#include <Reduce.hpp>
Public Member Functions | |
| ReduceParams (popops::Operation op, bool update, poplar::Tensor scale) | |
| Define the details of the reduce operation that will be performed by the reduce() and reduceWithOutput() functions. More... | |
Stores parameters for the reduce operation, as well as the basic operation being performed (for example, add or mul).
|
inline |
Define the details of the reduce operation that will be performed by the reduce() and reduceWithOutput() functions.
| op | The reduce operation to use. |
| scale | Can (optionally) scale the output. |
| update | Specify that the output should be updated, where out += reduce(in) rather than out = reduce(in). |