latest/doxygen/Embedding_8hpp_source.html

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

#ifndef popsparse_Embedding_hpp

#define popsparse_Embedding_hpp


#include <poplar/Graph.hpp>

#include <poplar/Interval.hpp>


#include <popsparse/FullyConnected.hpp>

#include <popsparse/FullyConnectedParams.hpp>

#include <popsparse/SparseTensor.hpp>


namespace popsparse {

namespace dynamic {


poplar::Tensor

createIndicesTensor(poplar::Graph &graph, const FullyConnectedParams &params,

                    std::size_t numIndices,

                    const poplar::OptionFlags &options = {},

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


poplar::Tensor createSliceTensor(poplar::Graph &graph,

                                 const poplar::Type &dataType,

                                 const FullyConnectedParams &params,

                                 std::size_t numIndices,

                                 const poplar::DebugContext &debugContext = {},

                                 const poplar::OptionFlags &options = {},

                                 PlanningCache *cache = nullptr);


poplar::Tensor embeddingSlice(poplar::Graph &graph, const SparseTensor &t,

                              const poplar::Tensor &indices,

                              poplar::program::Sequence &prog,

                              const FullyConnectedParams &params,

                              const poplar::DebugContext &debugContext = {},

                              const poplar::OptionFlags &options = {},

                              PlanningCache *cache = nullptr);


void embeddingUpdateAdd(

    poplar::Graph &graph, const SparseTensor &t, const poplar::Tensor &slices,

    const poplar::Tensor &indices, const poplar::Tensor &scale,

    poplar::program::Sequence &prog, const FullyConnectedParams &params,

    const poplar::DebugContext &debugContext = {},

    const poplar::OptionFlags &options = {}, PlanningCache *cache = nullptr);


} // end namespace dynamic


} // namespace popsparse


#endif // popsparse_Embedding_hpp

FullyConnectedParams.hpp
Parameters used for fully-connected layers using sparse tensors.

SparseTensor.hpp
Basic representation of a sparse tensor.

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::OptionFlags
A set of option/value string flags to be used in various APIs.
Definition: OptionFlags.hpp:24

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

poplar::Type
Class representing device data types.
Definition: Type.hpp:42

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

popsparse::dynamic::SparseTensor
Representation of a sparse tensor.
Definition: SparseTensor.hpp:16

popsparse::dynamic::createSliceTensor
poplar::Tensor createSliceTensor(poplar::Graph &graph, const poplar::Type &dataType, const FullyConnectedParams &params, std::size_t numIndices, const poplar::DebugContext &debugContext={}, const poplar::OptionFlags &options={}, PlanningCache *cache=nullptr)
Create and map a tensor to be updated from efficiently.

popsparse::dynamic::createIndicesTensor
poplar::Tensor createIndicesTensor(poplar::Graph &graph, const FullyConnectedParams &params, std::size_t numIndices, const poplar::OptionFlags &options={}, const poplar::DebugContext &debugContext={})
Create and map a tensor to contain indices for slicing/updating a tensor efficiently.

popsparse::dynamic::embeddingUpdateAdd
void embeddingUpdateAdd(poplar::Graph &graph, const SparseTensor &t, const poplar::Tensor &slices, const poplar::Tensor &indices, const poplar::Tensor &scale, poplar::program::Sequence &prog, const FullyConnectedParams &params, const poplar::DebugContext &debugContext={}, const poplar::OptionFlags &options={}, PlanningCache *cache=nullptr)
Update a sparse tensor with a set of slices at the given row indices.

popsparse::dynamic::embeddingSlice
poplar::Tensor embeddingSlice(poplar::Graph &graph, const SparseTensor &t, const poplar::Tensor &indices, poplar::program::Sequence &prog, const FullyConnectedParams &params, const poplar::DebugContext &debugContext={}, const poplar::OptionFlags &options={}, PlanningCache *cache=nullptr)
Take multiple slices from a base tensor.

popsparse
Support for sparse matrices.
Definition: codelets.hpp:8

FullyConnected.hpp
Fully-connected layers using sparse tensors.