Poplar and PopLibs
popops::ReduceParams Struct Reference

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

Detailed Description

Stores parameters for the reduce operation, as well as the basic operation being performed (for example, add or mul).

Constructor & Destructor Documentation

◆ ReduceParams()

popops::ReduceParams::ReduceParams ( popops::Operation  op,
bool  update,
poplar::Tensor  scale 
)
inline

Define the details of the reduce operation that will be performed by the reduce() and reduceWithOutput() functions.

Parameters
opThe reduce operation to use.
scaleCan (optionally) scale the output.
updateSpecify that the output should be updated, where out += reduce(in) rather than out = reduce(in).

The documentation for this struct was generated from the following file: