latest/doxygen/SpatialSoftMax_8hpp_source.html

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

#ifndef poplin_SpatialSoftMax_hpp

#define poplin_SpatialSoftMax_hpp


#include <poplar/Graph.hpp>

#include <popnn/NonLinearity.hpp>


namespace popnn {


std::pair<poplar::Tensor, poplar::Tensor>

spatialSoftMax2D(poplar::Graph &graph, poplar::program::Sequence &prog,

                 const poplar::Tensor &fields, float temperature,

                 bool disableSoftmax = false,

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


} // end namespace popnn


#endif // poplin_SpatialSoftMax_hpp

NonLinearity.hpp
Non-linearity operations.

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::spatialSoftMax2D
std::pair< poplar::Tensor, poplar::Tensor > spatialSoftMax2D(poplar::Graph &graph, poplar::program::Sequence &prog, const poplar::Tensor &fields, float temperature, bool disableSoftmax=false, const poplar::DebugContext &debugContext={})
Implements a spatial softmax specialised for 2D input fields.