Gather.hpp File Reference

Support for gather operations. More...

#include <poplar/Graph.hpp>
#include <poplar/Program.hpp>

Go to the source code of this file.

Classes
struct	popops::GatherParams
	Defines the parameters to a gather operation. More...

Namespaces
namespace	popops
	Common functions, such as elementwise and reductions.

Functions
poplar::Tensor	popops::createGatherInput (poplar::Graph &graph, const poplar::Type &type, const std::vector< std::size_t > &operandShape, unsigned axis, GatherParams params={}, const poplar::DebugContext &debugContext={})
	Create the input of the gather with only a single gather axis. More...
poplar::Tensor	popops::gather (poplar::Graph &graph, const poplar::Tensor &input, const poplar::Tensor &indices, unsigned axis, poplar::program::Sequence &prog, GatherParams params, const poplar::DebugContext &debugContext={})
	The gather operation stitches together several slices (each slice at a potentially different runtime offset) of an input tensor. More...
poplar::Tensor	popops::createGatherInput (poplar::Graph &graph, const poplar::Type &type, const std::vector< std::size_t > &inputShape, const std::vector< std::size_t > &sliceSizes, std::vector< unsigned > startIndexMap, const poplar::DebugContext &debugContext={})
	Create the input of the gather given a start index map. More...
poplar::Tensor	popops::gather (poplar::Graph &graph, const poplar::Tensor &input, const poplar::Tensor &indices, std::size_t indexVectorDim, const std::vector< std::size_t > &offsetDims, const std::vector< std::size_t > &sliceSizes, const std::vector< std::size_t > &collapsedSliceDims, const std::vector< unsigned > &startIndexMap, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={})
	The gather operation stitches together several slices (each slice at a potentially different runtime offset) of an input tensor. More...

Detailed Description

Support for gather operations.