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.