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.