Poplar and PopLibs
Encoding.hpp File Reference

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

Detailed Description

Encoding and generating ranges of integers.