Poplar and PopLibs
Scatter.hpp File Reference

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

Detailed Description

Scatter operations.