|
Poplar and PopLibs
|
Functions to fill tensors with values. More...
#include <poplar/Graph.hpp>#include <poplar/Program.hpp>#include <poplar/TypeTraits.hpp>Go to the source code of this file.
Namespaces | |
| namespace | popops |
| Common functions, such as elementwise and reductions. | |
Functions | |
| template<typename FillValueType > | |
| void | popops::fill (poplar::Graph &graph, poplar::Tensor t, const std::vector< poplar::Interval > &tileRegions, unsigned tile, poplar::ComputeSet fillCS, FillValueType fillValue) |
Appends vertices to fillCS which fills elements in tileRegions of t which reside on tile tile. More... | |
| template<typename FillValueType > | |
| void | popops::fill (poplar::Graph &graph, const poplar::Tensor &t, unsigned tile, poplar::ComputeSet fillCS, FillValueType fillValue) |
Appends vertices to fillCS which fills all elements of t which reside on tile tile. More... | |
| template<typename FillValueType > | |
| void | popops::fill (poplar::Graph &graph, const poplar::Tensor &t, const std::vector< std::vector< poplar::Interval > > &mapping, poplar::ComputeSet fillCS, FillValueType fillValue) |
Appends vertices to fillCS which fills elements in mapping of t which reside on tiles represented with mapping. More... | |
| template<typename FillValueType > | |
| void | popops::fill (poplar::Graph &graph, const poplar::Tensor &t, poplar::program::Sequence &prog, FillValueType fillValue, const poplar::DebugContext &debugContext={}) |
Appends programs to prog which fills all elements of the tensor t with a value of fillValue. More... | |
Functions to fill tensors with values.
Supported fillValue types are FLOAT, HALF, INT and UNSIGNED_INT.