latest/doxygen/Loss_8hpp_source.html

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

#ifndef popnn_Loss_hpp

#define popnn_Loss_hpp


namespace popnn {


enum LossType { SUM_SQUARED_LOSS, CROSS_ENTROPY_LOSS };


} // end namespace popnn


#ifndef __POPC__

#include "popops/EncodingConstants.hpp"

#include <poplar/Graph.hpp>

#include <poplar/Program.hpp>

#include <poplar/Tensor.hpp>


namespace popnn {


poplar::program::Program

calcLoss(poplar::Graph &graph, const poplar::Tensor &modelOutputs,

         const poplar::Tensor &expected, const poplar::Tensor &loss,

         const poplar::Tensor &deltas, const poplar::Tensor &deltasScale,

         const poplar::Tensor &modelOutputScaling, LossType lossType,

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


poplar::program::Program

calcLoss(poplar::Graph &graph, const poplar::Tensor &modelOutputs,

         const poplar::Tensor &expected, const poplar::Tensor &loss,

         const poplar::Tensor &deltas, LossType lossType,

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


poplar::program::Program

calcLoss(poplar::Graph &graph, const poplar::Tensor &modelOutputs,

         const poplar::Tensor &expected, const poplar::Tensor &loss,

         const poplar::Tensor &deltas, const poplar::Tensor &deltasScale,

         const poplar::Tensor &modelOutputScaling,

         const poplar::Tensor &numCorrect, LossType lossType,

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


poplar::program::Program

calcLoss(poplar::Graph &graph, const poplar::Tensor &modelOutputs,

         const poplar::Tensor &expected, const poplar::Tensor &loss,

         const poplar::Tensor &deltas, const poplar::Tensor &numCorrect,

         LossType lossType, const poplar::DebugContext &debugContext = {});


poplar::program::Program

calcAccuracy(poplar::Graph &graph, const poplar::Tensor &modelOutputs,

             const poplar::Tensor &expected, const poplar::Tensor &numCorrect,

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


poplar::Tensor argMax(poplar::Graph &graph, const poplar::Tensor &input,

                      poplar::program::Sequence &prog,

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


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

maxAndArgMax(poplar::Graph &graph, const poplar::Tensor &input,

             poplar::program::Sequence &prog,

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


poplar::Tensor argMin(poplar::Graph &graph, const poplar::Tensor &input,

                      poplar::program::Sequence &prog,

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


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

minAndArgMin(poplar::Graph &graph, const poplar::Tensor &input,

             poplar::program::Sequence &prog,

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


poplar::Tensor topK(poplar::Graph &graph, const poplar::Tensor &input,

                    poplar::Tensor &indices, unsigned K, bool sort,

                    poplar::program::Sequence &prog,

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


} // end namespace popnn


#endif // !__POPC__


#endif // popnn_Loss_hpp

EncodingConstants.hpp
Constants used by encoding functions.

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::Program
This class represents a control program that executes operations on the graph.
Definition: Program.hpp:30

poplar::program::Sequence
Program that executes a sequence of programs.
Definition: Program.hpp:77

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

popops::topK
poplar::Tensor topK(poplar::Graph &graph, poplar::program::Sequence &prog, const poplar::Tensor &t, const TopKParams &params, const poplar::DebugContext &debugContext={})
Return the top k values in the innermost dimension of a tensor.

popops::sort
poplar::Tensor sort(poplar::Graph &graph, const poplar::Tensor &t, unsigned dim, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={})