Poplar and PopLibs
LogSoftmax.hpp File Reference

Log of softmax functions. More...

#include <poplar/Graph.hpp>
#include <poplar/Program.hpp>

Go to the source code of this file.

Namespaces

namespace  popnn
 Functions used in neural networks.
 

Functions

void popnn::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. More...
 
poplar::Tensor popnn::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. More...
 

Detailed Description

Log of softmax functions.