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