Poplar and PopLibs
|
Functions to provide counted loops of programs. More...
#include <poplar/Graph.hpp>
#include <poplar/Program.hpp>
#include <poputil/DebugInfo.hpp>
#include <poputil/VertexTemplates.hpp>
Go to the source code of this file.
Namespaces | |
namespace | popops |
Common functions, such as elementwise and reductions. | |
Functions | |
poplar::program::Sequence | popops::countedLoop (poplar::Graph &graph, std::size_t begin, std::size_t end, size_t step, const CountedLoopBodyType &body, const poplar::DebugContext &debugContext={}) |
Create a loop program with constant initial count, increment and end value. More... | |
poplar::program::Sequence | popops::countedLoop (poplar::Graph &graph, std::size_t count, const CountedLoopBodyType &body, const poplar::DebugContext &debugContext={}) |
Create a loop program which executes count times. More... | |
poplar::program::Sequence | popops::countedForLoop (poplar::Graph &graph, const poplar::Tensor &count, int initialCount, const poplar::Tensor &countLimit, int countStep, const poplar::program::Program &body, const poplar::DebugContext &debugContext={}) |
Create a for-loop program with constant initial count and increment, and a tensor as the end value. More... | |
poplar::program::Sequence | popops::countedForLoop (poplar::Graph &graph, int initialCount, const poplar::Tensor &countLimit, int countStep, const poplar::program::Program &body, const poplar::DebugContext &debugContext={}) |
Create a for loop program with constant initial count and increment and a tensor as the end value. More... | |
Functions to provide counted loops of programs.