ScaledAdd.hpp File Reference

Functions for scaling and adding tensors. More...

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

Go to the source code of this file.

Namespaces
namespace	popops
	Common functions, such as elementwise and reductions.

Functions
void	popops::scaledAddTo (poplar::Graph &graph, poplar::Tensor A, poplar::Tensor B, float scaleB, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={}, const poplar::OptionFlags &options={})
	Add the elements of one tensor multiplied by a scalar to another tensor. More...
void	popops::scaledAddTo (poplar::Graph &graph, poplar::Tensor A, poplar::Tensor B, poplar::Tensor scaleB, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={}, const poplar::OptionFlags &options={})
	Add the elements of one tensor each multiplied by a (scalar) tensor to another tensor. More...
void	popops::scaledSubtractFrom (poplar::Graph &graph, poplar::Tensor A, poplar::Tensor B, float scaleB, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={}, const poplar::OptionFlags &options={})
	Subtract the elements of one tensor multiplied by a scalar from another tensor. More...
void	popops::scaledSubtractFrom (poplar::Graph &graph, poplar::Tensor A, poplar::Tensor B, poplar::Tensor scaleB, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={}, const poplar::OptionFlags &options={})
	Subtract the elements of one tensor each multiplied by a (scalar) tensor from another tensor. More...
void	popops::scaledAddTo (poplar::Graph &graph, poplar::Tensor A, poplar::Tensor scaleA, poplar::Tensor B, poplar::Tensor scaleB, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={}, const poplar::OptionFlags &options={})
	Scale the elements of one tensor and add the scaled elements of another tensor to it. More...
void	popops::scaledAddTo (poplar::Graph &graph, poplar::Tensor A, poplar::Tensor scaleA, poplar::Tensor B, poplar::Tensor scaleB, poplar::program::Sequence &prog, const ScaledAddSpecialisation speciality, const poplar::DebugContext &debugContext={}, const poplar::OptionFlags &options={})
	Scale the elements of one tensor and add the scaled elements of another tensor to it. More...
void	popops::scaledAddTo (poplar::Graph &graph, poplar::Tensor A, float scaleA, poplar::Tensor B, float scaleB, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={}, const poplar::OptionFlags &options={})
	Scale the elements of one tensor and add the scaled elements of another tensor to it. More...
void	popops::scaledAddTo (poplar::Graph &graph, poplar::Tensor A, float scaleA, poplar::Tensor B, float scaleB, poplar::program::Sequence &prog, const ScaledAddSpecialisation speciality, const poplar::DebugContext &debugContext={}, const poplar::OptionFlags &options={})
	Scale the elements of one tensor and add the scaled elements of another tensor to it. More...
void	popops::scaledSubtractFrom (poplar::Graph &graph, poplar::Tensor A, poplar::Tensor scaleA, poplar::Tensor B, poplar::Tensor scaleB, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={}, const poplar::OptionFlags &options={})
	Scale the elements of one tensor and subtract the scaled elements of another tensor to it. More...
void	popops::scaledSubtractFrom (poplar::Graph &graph, poplar::Tensor A, float scaleA, poplar::Tensor B, float scaleB, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={}, const poplar::OptionFlags &options={})
	Scale the elements of one tensor and subtract the scaled elements of another tensor to it. More...

Detailed Description

Functions for scaling and adding tensors.