latest/doxygen/NormaliseImage_8hpp_source.html

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

#ifndef popops_NormaliseImage_hpp

#define popops_NormaliseImage_hpp

#include <poplar/Graph.hpp>

#include <poplar/Program.hpp>


namespace popops {


poplar::Tensor

createNormaliseImageInput(poplar::Graph &graph, const poplar::Type &type,

                          const poplar::ArrayRef<std::size_t> shape,

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


poplar::Tensor normaliseImage(poplar::Graph &graph,

                              poplar::program::Sequence &seq,

                              poplar::Tensor tIn, float inScale,

                              poplar::Tensor offsets, poplar::Tensor scales,

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


} // namespace popops


#endif // popops_NormaliseImage_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::Tensor
A reference to a subset of tensor elements.
Definition: Tensor.hpp:38

poplar::Type
Class representing device data types.
Definition: Type.hpp:42

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::normaliseImage
poplar::Tensor normaliseImage(poplar::Graph &graph, poplar::program::Sequence &seq, poplar::Tensor tIn, float inScale, poplar::Tensor offsets, poplar::Tensor scales, const poplar::DebugContext &debugContext={})
Pad a tensor to have 4 channel dimensions.

popops::createNormaliseImageInput
poplar::Tensor createNormaliseImageInput(poplar::Graph &graph, const poplar::Type &type, const poplar::ArrayRef< std::size_t > shape, const poplar::DebugContext &debugContext={})
Add a tensor for a 3-channel image suitable for padding to 4 channels.