latest/doxygen/Encoding_8hpp_source.html

// Copyright (c) 2018 Graphcore Ltd. All rights reserved.

#ifndef popops_Encoding_hpp

#define popops_Encoding_hpp


#include "EncodingConstants.hpp"

#include "poplar/Graph.hpp"

#include "poplar/Tensor.hpp"

#include "poplar/Type.hpp"

#include <string>


namespace popops {


void encodeOneHot(poplar::Graph &graph, const poplar::Tensor &indices,

                  const poplar::Tensor &encoded,

                  poplar::program::Sequence &prog,

                  const poplar::DebugContext &debugContext = {});


void 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 = {});


void iota(poplar::Graph &graph, const poplar::Tensor &t, unsigned startInteger,

          poplar::program::Sequence &prog,

          const poplar::DebugContext &debugContext = {});


void iota(poplar::Graph &graph, const poplar::Tensor &t, int startInteger,

          poplar::program::Sequence &prog,

          const poplar::DebugContext &debugContext = {});


} // end namespace popops


#endif // popops_Encoding_hpp

EncodingConstants.hpp
Constants used by encoding functions.

poplar::DebugContext
DebugContext gathers the common external parameters of the context of an operation.
Definition: DebugContext.hpp:221

poplar::Graph
This class represents a graph program to be executed on the IPU.
Definition: Graph.hpp:52

poplar::Tensor
A reference to a subset of tensor elements.
Definition: Tensor.hpp:38

poplar::program::Sequence
Program that executes a sequence of programs.
Definition: Program.hpp:77

popops
Common functions, such as elementwise and reductions.
Definition: AllTrue.hpp:15

popops::encodeOneHot
void 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.

popops::iota
void 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),...