|
Poplar and PopLibs
|
Scatter operations. More...
#include <poplar/Graph.hpp>#include <poplar/Program.hpp>Go to the source code of this file.
Namespaces | |
| namespace | popops |
| Common functions, such as elementwise and reductions. | |
Functions | |
| void | popops::scatter (poplar::Graph &graph, const poplar::Tensor &operand, const poplar::Tensor &indices, const poplar::Tensor &updates, std::size_t indexVectorDim, std::vector< unsigned > updateWindowDims, std::vector< std::size_t > insertWindowDims, std::vector< unsigned > scatterDimsToOperandDims, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={}) |
The scatter operation generates a result which is the value of the input array operand, with several slices (at indices specified by indices) updated with the values in updates. More... | |
| void | popops::scatter (poplar::Graph &graph, const poplar::Tensor &operand, const poplar::Tensor &indices, const poplar::Tensor &updates, std::size_t indexVectorDim, std::vector< unsigned > updateWindowDims, std::vector< std::size_t > insertWindowDims, std::vector< unsigned > scatterDimsToOperandDims, UpdateComputationFunc &updateComputation, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={}) |
| Similar to the above scatter(), but allows for a user defined update computation. More... | |
Scatter operations.