Poplar and PopLibs
|
Encoding and generating ranges of integers. More...
#include "EncodingConstants.hpp"
#include "poplar/Graph.hpp"
#include "poplar/Tensor.hpp"
#include "poplar/Type.hpp"
#include <string>
Go to the source code of this file.
Namespaces | |
namespace | popops |
Common functions, such as elementwise and reductions. | |
Functions | |
void | popops::encodeOneHot (poplar::Graph &graph, const poplar::Tensor &indices, const poplar::Tensor &encoded, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={}) |
Encode a given set of indices as a set of one-hot vectors per-index with a hot element at that index. More... | |
void | popops::encodeOneHot (poplar::Graph &graph, const poplar::Tensor &indices, const poplar::Tensor &encoded, poplar::program::Sequence &prog, const poplar::Tensor &on, const poplar::Tensor &off, const poplar::DebugContext &debugContext={}) |
Encode a given set of indices as a set of one-hot vectors per-index with a hot element at that index. More... | |
void | popops::iota (poplar::Graph &graph, const poplar::Tensor &t, unsigned startInteger, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={}) |
Fill a tensor with a right-open range of unsigned integers: [startInteger, startInteger + length), where length is the number of elements in the mapped 1-D output tensor t . More... | |
void | popops::iota (poplar::Graph &graph, const poplar::Tensor &t, int startInteger, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={}) |
Fill a tensor with a right-open range of signed integers: [startInteger, startInteger + length), where length is the number of elements in the mapped 1-D output tensor t . More... | |
Encoding and generating ranges of integers.