Poplar and PopLibs
NormaliseImage.hpp File Reference

Functions for padding and normalising image tensors. More...

#include <poplar/Graph.hpp>
#include <poplar/Program.hpp>

Go to the source code of this file.

Namespaces

namespace  popops
 Common functions, such as elementwise and reductions.
 

Functions

poplar::Tensor popops::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. More...
 
poplar::Tensor popops::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. More...
 

Detailed Description

Functions for padding and normalising image tensors.