latest/doxygen/GatherStatistics_8hpp_source.html

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

#ifndef popops_GatherStatistics_hpp

#define popops_GatherStatistics_hpp


#include <poplar/Graph.hpp>

#include <poplar/OptionFlags.hpp>

#include <poplar/Program.hpp>

#include <string>


namespace popops {


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

                         const poplar::Tensor &levels, bool absoluteOfInput,

                         poplar::program::Sequence &prog,

                         const poplar::DebugContext &debugContext = {},

                         const poplar::OptionFlags &options = {});


void histogram(poplar::Graph &graph, const poplar::Tensor &input,

               poplar::Tensor &output, bool updateOutput,

               const poplar::Tensor &levels, bool absoluteOfInput,

               poplar::program::Sequence &prog,

               const poplar::DebugContext &debugContext = {},

               const poplar::OptionFlags &options = {});

} // namespace popops


#endif // popops_GatherStatistics_hpp

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::OptionFlags
A set of option/value string flags to be used in various APIs.
Definition: OptionFlags.hpp:24

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

popops
Common functions, such as elementwise and reductions.
Definition: AllTrue.hpp:15

popops::histogram
poplar::Tensor histogram(poplar::Graph &graph, const poplar::Tensor &input, const poplar::Tensor &levels, bool absoluteOfInput, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={}, const poplar::OptionFlags &options={})
Gather a histogram representing the statistics of the input tensor.