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