latest/doxygen/LogSoftmax_8hpp_source.html

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

#ifndef popnn_LogSoftmax_hpp

#define popnn_LogSoftmax_hpp


#include <poplar/Graph.hpp>

#include <poplar/Program.hpp>


namespace popnn {


void logSoftmaxInPlace(poplar::Graph &graph, poplar::Tensor t,

                       poplar::program::Sequence &prog,

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


poplar::Tensor logSoftmax(poplar::Graph &graph, poplar::Tensor t,

                          poplar::program::Sequence &prog,

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


} // end namespace popnn


#endif // popnn_LogSoftmax_hpp

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

popnn
Functions used in neural networks.
Definition: BatchNorm.hpp:14

popnn::logSoftmaxInPlace
void logSoftmaxInPlace(poplar::Graph &graph, poplar::Tensor t, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={})
Update tensor t by computing log of softmax in-place.

popnn::logSoftmax
poplar::Tensor logSoftmax(poplar::Graph &graph, poplar::Tensor t, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={})
Compute the log of the softmax to tensor t and return the result.