Cast.hpp File Reference

Casts between tensor types. More...

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

Go to the source code of this file.

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

Functions
poplar::Tensor	popops::cast (poplar::Graph &graph, const poplar::Tensor &src, const poplar::Type &dstType, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={})
	Cast elements of the specified src tensor to dstType, returning the result as a new tensor. More...
poplar::Tensor	popops::cast (poplar::Graph &graph, const poplar::Tensor &src, const poplar::Type &dstType, const poplar::Tensor &metadata, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={})
	Cast elements of the specified src tensor to dstType, returning the result as a new tensor. More...
poplar::program::Program	popops::cast (poplar::Graph &graph, poplar::Tensor src, poplar::Tensor dst, const poplar::DebugContext &debugContext={})
	Create a program to copy tensor casting between types (for example, half->float). More...
void	popops::cast (poplar::Graph &graph, poplar::Tensor src, poplar::Tensor dst, poplar::ComputeSet cs)
	Create vertices to copy element wise from the src tensor to the dst tensor casting between types (for example, half->float). More...
poplar::Tensor	popops::cast (poplar::Graph &graph, poplar::Tensor src, const poplar::Type &dstType, poplar::ComputeSet cs, const poplar::DebugContext &debugContext={})
	Create vertices to cast elements of the specified src tensor to dstType, returning the result as a new tensor. More...
poplar::Tensor	popops::cast (poplar::Graph &graph, poplar::Tensor src, const poplar::Type &dstType, const poplar::Tensor &metadata, poplar::ComputeSet cs, const poplar::DebugContext &debugContext={})
	Create vertices to cast elements of the specified src tensor to dstType, returning the result as a new tensor. More...
poplar::Tensor	popops::checkAccuracyWhenCast (poplar::Graph &graph, const poplar::Tensor &input, poplar::Type outputType, double tolerance, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={})
	Helper function which checks the relative error in the tensor input when casting it to type outputType. More...

Detailed Description

Casts between tensor types.