Poplar and PopLibs
Fill.hpp File Reference

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

Detailed Description

Functions to fill tensors with values.

Supported fillValue types are FLOAT, HALF, INT and UNSIGNED_INT.