|
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.