Poplar and PopLibs
|
Set elements of tensor to zero. More...
#include "poplar/Graph.hpp"
#include "poplar/Program.hpp"
#include <vector>
Go to the source code of this file.
Namespaces | |
namespace | popops |
Common functions, such as elementwise and reductions. | |
Functions | |
void | popops::zero (poplar::Graph &graph, poplar::Tensor t, const std::vector< poplar::Interval > &tileRegions, unsigned tile, poplar::ComputeSet zeroCS) |
Append vertices to compute set zeroCS which zeroes elements in tileRegions of tensor t which reside on tile . More... | |
void | popops::zero (poplar::Graph &graph, const poplar::Tensor &t, unsigned tile, poplar::ComputeSet zeroCS) |
Append vertices to compute set zeroCS which zeroes all elements of tensor t which reside on tile . More... | |
void | popops::zero (poplar::Graph &graph, const poplar::Tensor &t, const std::vector< std::vector< poplar::Interval > > &mapping, poplar::ComputeSet zeroCS) |
Append vertices to compute set zeroCS which zeroes elements in mapping of tensor t which reside on tiles represented with mapping . More... | |
void | popops::zero (poplar::Graph &graph, const poplar::Tensor &t, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={}) |
Append programs to program sequence prog which zeroes all elements of tensor t . More... | |
Set elements of tensor to zero.