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.