|
Poplar and PopLibs
|
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... | |
Log of softmax functions.