Poplar and PopLibs
|
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... | |
Functions for padding and normalising image tensors.