Expr
#include <popops/Expr.hpp>
Expressions with elements of tensors.
Defines
-
POPLIBS_DEFINE_EXPR_UNARY_OP(Name, Op)
-
POPLIBS_DEFINE_EXPR_UNARY_OP_AND_SYMBOL(Name, Op, Sym)
-
POPLIBS_DEFINE_EXPR_BINARY_OP(Name, Op)
-
POPLIBS_DEFINE_EXPR_BINARY_OP_AND_SYMBOL(Name, Op, Sym)
-
POPLIBS_DEFINE_EXPR_TERNARY_OP(Name, Op)
-
namespace popops
Common functions, such as elementwise and reductions.
map
Map an expression across tensors.
Elementwise Options
enableGenerateCodelet(true, false) [=true]When true and the following conditions are met, poplar will generate a codelet to execute the map operation. Otherwise, it will sequence poplibs codelets to create the expression.
All of the inputs are of the same size
Inputs do not alias
Multiple operations are being performed
mapInPlace
Update the input tensors with the result of map().
mapWithOutput
Write the result of map() to the given output tensor.
checkTypes
Check that the host compile-time type
constTypeis compatible with the run-time IPU typeelementType.- param elementType
The run-time IPU type.
- param constant
Unused.
- tparam constType
The host compile-time type.
- throws std::runtime_error
If the types are not compatible.
varianceToInvStdDev
Convert variance to inverse standard deviation.
Each element in the output tensor is the result of
1 / sqrt(variance_value + epsilon), wherevariance_valueis the corresponding element invariance.Warning
If
variance_value + epsilonis zero then the result will be invalid and this operation could generate a divide-by-zero floating-point exception (if enabled).- param graph
The graph to update.
- param variance
A tensor of variance values.
- param epsilon
A (typically small) scalar to add to the variance values, to avoid numerical issues (for example, divide by zero).
- param prog
The sequence of programs to append this conversion operation to.
- param debugContext
Optional debug information.
- return
A tensor where each element is the inverse standard deviation.
invStdDevToVariance
Convert inverse standard deviation to variance.
Each element in the output tensor is the result of
1 / (invStdDev_value + epsilon)^2, whereinvStdDev_valueis the corresponding element ininvStdDev.Warning
If
invStdDev_value + epsilonis zero then the result will be invalid and this operation could generate a divide-by-zero floating-point exception (if enabled).- param graph
The graph to update.
- param invStdDev
A tensor of inverse standard deviation values.
- param epsilon
A (typically small) scalar to add to the variance values, to avoid numerical issues (for example, divide by zero).
- param prog
The sequence of programs to append this conversion operation to.
- param debugContext
Optional debug information.
- param options
A list of flags to pass to the expression evaluator.
- return
A tensor where each element is the variance.
add
Add each element in
Ato the corresponding element inB.- param graph
The graph to update.
- param A
A tensor of elements.
- param B
A tensor of elements.
- param prog
The sequence to extend with the execution of the expression evaluation.
- param debugContext
Optional debug information
- param options
Element-wise options. See map().
- return
A tensor where each element is the result of
a + b, whereaandbare the corresponding elements ofAandBtensors respectively.
addInPlace
Update the tensor
Awith the result of add().See add() for parameter descriptions.
addWithOutput
Write the result of add() to the given output tensor,
out.See add() for the remaining parameter descriptions.
- param out
The tensor to write the results to.
sub
Subtract the elements of
BfromAand return the result in a new tensor.- param graph
The graph to update.
- param A
The tensor of elements which will be subtracted from.
- param B
The tensor of elements to subtract from
A.- param prog
The sequence to extend with the execution of the expression evaluation.
- param debugContext
Optional debug information.
- param options
Element-wise options. See map().
- return
A tensor where each element is equal to a - b, where
aandbare the corresponding elements ofAandBtensors respectively.
subInPlace
Update the tensor
Awith the result of sub().See sub() for parameter descriptions.
subWithOutput
Write the result of sub() to the given output tensor,
out.See sub() for the remaining parameter descriptions.
- param out
The tensor to write the results to.
mul
Multiply each element in
Aby the corresponding element inB.- param graph
The graph to update.
- param A
A tensor of elements.
- param B
A tensor of elements.
- param prog
The sequence to extend with the execution of the expression evaluation.
- param debugContext
Optional debug information.
- param options
Element-wise options. See map().
- return
A tensor where each element is the result of
a * b, whereaandbare the corresponding elements ofAandBtensors respectively.
mulInPlace
Update the tensor
Awith the result of mul().See mul() for parameter descriptions.
mulWithOutput
Write the result of mul() to the given output tensor,
out.See mul() for the remaining parameter descriptions.
- param out
The tensor to write the results to.
div
Divide each element in
Aby the corresponding element inB.- param graph
The graph to update.
- param A
The tensor of dividends.
- param B
The tensor of divisors.
- param prog
The sequence to extend with the execution of the expression evaluation.
- param debugContext
Optional debug information
- param options
Element-wise options. See map().
- return
A tensor where each element is the result of
a / b, whereaandbare the corresponding elements ofAandBtensors respectively.
divInPlace
Update the tensor
Awith the result of div().See div() for parameter descriptions.
divWithOutput
Write the result of div() to the given output tensor,
out.See div() for the remaining parameter descriptions.
- param out
The tensor to write the results to.
pow
Compute each element in
Ato the power of the corresponding element inB.- param graph
The graph to update.
- param A
The tensor of bases.
- param B
The tensor of exponents.
- param prog
The sequence to extend with the execution of the expression evaluation.
- param debugContext
Optional debug information.
- param options
Element-wise options. See map().
- return
A tensor where each element is equal to
pow(a, b), whereaandbare the corresponding elements ofAandBtensors respectively.
powInPlace
Update the tensor
Awith the result of pow().See pow() for parameter descriptions.
powWithOutput
Write the result of pow() to the given output tensor,
out.See pow() for the remaining parameter descriptions.
- param out
The tensor to write the results to.
rem
Compute the remainder of each element in
Adivided by the corresponding element inB.- param graph
The graph to update.
- param A
The tensor of dividends.
- param B
The tensor of divisors.
- param prog
The sequence to extend with the execution of the expression evaluation.
- param debugContext
Optional debug information.
- param options
Element-wise options. See map().
- return
A tensor where each element is equal to a % b, where
aandbare the corresponding elements ofAandBtensors respectively.
remInPlace
Update the tensor
Awith the result of rem().See rem() for parameter descriptions.
remWithOutput
Write the result of rem() to the given output tensor,
out.See rem() for the remaining parameter descriptions.
- param out
The tensor to write the results to.
bitwiseAnd
Compute the bitwise AND of each element in
Awith the corresponding element inB.- param graph
The graph to update.
- param A
A tensor of elements.
- param B
A tensor of elements.
- param prog
The sequence to extend with the execution of the expression evaluation.
- param debugContext
Optional debug information
- param options
Element-wise options. See map().
- return
A tensor where each element is the result of
a & b, whereaandbarethe corresponding elements ofAandBtensors respectively.
bitwiseAndInPlace
Update the tensor
Awith the result of bitwiseAnd().See bitwiseAnd() for parameter descriptions.
bitwiseAndWithOutput
Write the result of bitwiseAnd() to the given output tensor,
out.See bitwiseAnd() for the remaining parameter descriptions.
- param out
The tensor to write the results to.
bitwiseOr
Compute the bitwise OR of each element in
Awith the corresponding element inB.- param graph
The graph to update.
- param A
A tensor of elements.
- param B
A tensor of elements.
- param prog
The sequence to extend with the execution of the expression evaluation.
- param debugContext
Optional debug information
- param options
Element-wise options. See map().
- return
A tensor where each element is the result of
a | b, whereaandbare the corresponding elements ofAandBtensors respectively.
bitwiseOrInPlace
Update the tensor
Awith the result of bitwiseOr().See bitwiseOr() for parameter descriptions.
bitwiseOrWithOutput
Write the result of bitwiseOr() to the given output tensor,
out.See bitwiseOr() for the remaining parameter descriptions.
- param out
The tensor to write the results to.
bitwiseXor
Compute the bitwise XOR of each element in
Awith the corresponding element inB.- param graph
The graph to update.
- param A
A tensor of elements.
- param B
A tensor of elements.
- param prog
The sequence to extend with the execution of the expression evaluation.
- param debugContext
Optional debug information
- param options
Element-wise options. See map().
- return
A tensor where each element is the result of
a ^ b, whereaandbare the corresponding elements ofAandBtensors respectively.
bitwiseXorInPlace
Update the tensor
Awith the result of bitwiseXor().See bitwiseXnor() for parameter descriptions.
bitwiseXorWithOutput
Write the result of bitwiseXor() to the given output tensor,
out.See bitwiseXor() for the remaining parameter descriptions.
- param out
The tensor to write the results to.
bitwiseXnor
Compute the bitwise XNOR of each element in
Awith the corresponding element inB.- param graph
The graph to update.
- param A
A tensor of elements.
- param B
A tensor of elements.
- param prog
The sequence to extend with the execution of the expression evaluation.
- param debugContext
Optional debug information
- param options
Element-wise options. See map().
- return
A tensor where each element is the result of
!(a ^ b), whereaandbare the corresponding elements ofAandBtensors respectively.
bitwiseXnorInPlace
Update the tensor
Awith the result of bitwiseXnor().See bitwiseXnor() for parameter descriptions.
bitwiseXnorWithOutput
Write the result of bitwiseXnor() to the given output tensor,
out.See bitwiseXnor() for the remaining parameter descriptions.
- param out
The tensor to write the results to.
shiftLeft
Shift the elements of
Aleft by the corresponding elements ofB.- param graph
The graph to update.
- param A
The tensor of elements which to left-shift.
- param B
The tensor of elements that describe the amount to left-shift
Aby.- param prog
The sequence to extend with the execution of the expression evaluation.
- param debugContext
Optional debug information.
- param options
Element-wise options. See map().
- return
A tensor where each element is equal to a << b, where
aandbare the corresponding elements ofAandBtensors respectively.
shiftLeftInPlace
Update the tensor
Awith the result of shiftLeft().See shiftLeft() for parameter descriptions.
shiftLeftWithOutput
Write the result of shiftLeft() to the given output tensor,
out.See shiftLeft() for the remaining parameter descriptions.
- param out
The tensor to write the results to.
shiftRight
Shift the elements of
Aright by the corresponding elements ofB.- param graph
The graph to update.
- param A
The tensor of elements which to right-shift.
- param B
The tensor of elements that describe the amount to right-shift by.
A.- param prog
The sequence to extend with the execution of the expression evaluation.
- param debugContext
Optional debug information.
- param options
Element-wise options. See map().
- return
A tensor where each element is equal to a >> b (without sign extension), where
aandbare the corresponding elements ofAandBtensors respectively.
shiftRightInPlace
Update the tensor
Awith the result of shiftRight().See shiftRight() for parameter descriptions.
shiftRightWithOutput
Write the result of shiftRight() to the given output tensor,
out.See shiftRight() for the remaining parameter descriptions.
- param out
The tensor to write the results to.
shiftRightSignExtend
Shift the elements of
Aright with sign extension by the corresponding elements ofB.- param graph
The graph to update.
- param A
The tensor of elements which to right-shift.
- param B
The tensor of elements that describe the amount to right-shift
Aby.- param prog
The sequence to extend with the execution of the expression evaluation.
- param debugContext
Optional debug information.
- param options
Element-wise options. See map().
- return
A tensor where each element is equal to
a >> bwith sign extension, whereaandbare the corresponding elements ofAandBtensors respectively.
shiftRightSignExtendInPlace
Update the tensor
Awith the result of shiftRightSignExtend().See shiftRightSignExtend() for parameter descriptions.
shiftRightSignExtendWithOutput
Write the result of shiftRightSignExtend() to the given output tensor,
out.See shiftRightSignExtend() for the remaining parameter descriptions.
- param out
The tensor to write the results to.
logicalAnd
Compute the logical AND (
&&) of each element inAwith the corresponding element inB.- param graph
The graph to update.
- param A
A tensor of elements.
- param B
A tensor of elements.
- param prog
The sequence to extend with the execution of the expression evaluation.
- param debugContext
Optional debug information.
- param options
Element-wise options. See map().
- return
A tensor where each element is the result of
a && b, whereaandbare the corresponding elements ofAandBtensors respectively.
logicalAndInPlace
Update the tensor
Awith the result of logicalAnd().See logicalAnd() for parameter descriptions.
logicalAndWithOutput
Write the result of logicalAnd() to the given output tensor,
out.See logicalAnd() for the remaining parameter descriptions.
- param out
The tensor to write the booleans to.
logicalOr
Compute the logical OR (
||) of each element inAwith the corresponding element inB.- param graph
The graph to update.
- param A
A tensor of elements.
- param B
A tensor of elements.
- param prog
The sequence to extend with the execution of the expression evaluation.
- param debugContext
Optional debug information.
- param options
Element-wise options. See map().
- return
A tensor where each element is the result of
a || b, whereaandbare the corresponding elements ofAandBtensors respectively.
logicalOrInPlace
Update the tensor
Awith the result of logicalOr().See logicalOr() for parameter descriptions.
logicalOrWithOutput
Write the result of logicalOr() to the given output tensor,
out.See logicalOr() for the remaining parameter descriptions.
- param out
The tensor to write the booleans to.
eq
Check if each element in
Ais equal to the corresponding element inB.- param graph
The graph to update.
- param A
A tensor of elements.
- param B
A tensor of elements.
- param prog
The sequence to extend with the execution of the expression evaluation.
- param debugContext
Optional debug information
- param options
Element-wise options. See map().
- return
A tensor where each element is the result of
a == b, whereaandbare the corresponding elements ofAandBtensors respectively.
eqInPlace
Update the tensor
Awith the result of eq().See eq() for parameter descriptions.
eqWithOutput
Write the result of eq() to the given output tensor,
out.See eq() for the remaining parameter descriptions.
- param out
The tensor to write the booleans to.
neq
Check if each element in
Ais not equal to the corresponding element inB.- param graph
The graph to update.
- param A
A tensor of elements.
- param B
A tensor of elements.
- param prog
The sequence to extend with the execution of the expression evaluation.
- param debugContext
Optional debug information.
- param options
Element-wise options. See map().
- return
A tensor where each element is the result of
a != b, whereaandbare the corresponding elements ofAandBtensors respectively.
neqInPlace
Update the tensor
Awith the result of neq().See neq() for parameter descriptions.
neqWithOutput
Write the result of neq() to the given output tensor,
out.See neq() for the remaining parameter descriptions.
- param out
The tensor to write the booleans to.
gteq
Check if each element in
Ais greater than or equal to the corresponding element inB.- param graph
The graph to update.
- param A
A tensor of elements.
- param B
A tensor of elements.
- param prog
The sequence to extend with the execution of the expression evaluation.
- param debugContext
Optional debug information
- param options
Element-wise options. See map().
- return
A tensor where each element is the result of
a >= b, whereaandbare the corresponding elements ofAandBtensors respectively.
gteqInPlace
Update the tensor
Awith the result of gteq().See gteq() for parameter descriptions.
gteqWithOutput
Write the result of gteq() to the given output tensor,
out.See gteq() for the remaining parameter descriptions.
- param out
The tensor to write the booleans to.
gt
Check if each element in
Ais greater than the corresponding element inB.- param graph
The graph to update.
- param A
A tensor of elements.
- param B
A tensor of elements.
- param prog
The sequence to extend with the execution of the expression evaluation.
- param debugContext
Optional debug information
- param options
Element-wise options. See map().
- return
A tensor where each element is the result of
a > b, whereaandbare the corresponding elements ofAandBtensors respectively.
gtInPlace
Update the tensor
Awith the result of gt().See gt() for parameter descriptions.
gtWithOutput
Write the result of gt() to the given output tensor,
out.See gt() for the remaining parameter descriptions.
- param out
The tensor to write the booleans to.
lteq
Check if each element in
Ais less than or equal to the corresponding element inB.- param graph
The graph to update.
- param A
A tensor of elements.
- param B
A tensor of elements.
- param prog
The sequence to extend with the execution of the expression evaluation.
- param debugContext
Optional debug information.
- param options
Element-wise options. See map().
- return
A tensor where each element is the result of
a <= b, whereaandbare the corresponding elements ofAandBtensors respectively.
lteqInPlace
Update the
Awith the result of lteq().See lteq() for parameter descriptions.
lteqWithOutput
Write the result of lteq() to the given output tensor,
out.See lteq() for the remaining parameter descriptions.
- param out
The tensor to write the booleans to.
lt
Check if each element in
Ais less than the corresponding element inB.- param graph
The graph to update.
- param A
A tensor of elements.
- param B
A tensor of elements.
- param prog
The sequence to extend with the execution of the expression evaluation.
- param debugContext
Optional debug information.
- param options
Element-wise options. See map().
- return
A tensor where each element is the result of
a < b, whereaandbare the corresponding elements ofAandBtensors respectively.
ltInPlace
Update the
Awith the result of lt().See lt() for parameter descriptions.
ltWithOutput
Write the result of lt() to the given output tensor,
out.See lt() for the remaining parameter descriptions.
- param out
The tensor to write the boolean results to.
max
Compute the maximum of each element in
Awith the corresponding element inB.- param graph
The graph to update.
- param A
A tensor of elements.
- param B
A tensor of elements.
- param prog
The sequence to extend with the execution of the expression evaluation.
- param debugContext
Optional debug information.
- param options
Element-wise options. See map().
- return
A tensor where each element is the result of
max(a, b), whereaandbare the corresponding elements ofAandBtensors respectively.
maxWithOutput
Write the result of max() to the given output tensor,
out.See max() for the remaining parameter descriptions.
- param out
The tensor to write the maximums to.
min
Compute the minimum of each element in
Awith the corresponding element inB.- param graph
The graph to update.
- param A
A tensor of elements.
- param B
A tensor of elements.
- param prog
The sequence to extend with the execution of the expression evaluation.
- param debugContext
Optional debug information.
- param options
Element-wise options. See map().
- return
A tensor where each element is the result of
min(a, b), whereaandbare the corresponding elements ofAandBtensors respectively.
minWithOutput
Write the result of min() to the given output tensor,
out.See min() for the remaining parameter descriptions.
- param out
The tensor to write the minimums to.
atan2
Compute the element-wise arctangent of
A / B.- param graph
The graph to update.
- param A
A tensor of elements.
- param B
A tensor of elements.
- param prog
The sequence to extend with the execution of the expression evaluation.
- param debugContext
Optional debug information
- param options
Element-wise options. See map().
- return
A tensor where each element is the result of
arctan(a / b);aandbare the corresponding elements ofAandBtensors respectively.
atan2InPlace
Update the tensor
Awith the result of atan2().See atan2() for parameter descriptions.
atan2WithOutput
Write the result of atan2() to the given output tensor,
out.See atan2() for the remaining parameter descriptions.
- param out
The tensor to write the result to.
invStdDevToVarianceInPlace
Update the
invStdDevtensor with the result of invStdDevToVariance().See invStdDevToVariance() for parameter descriptions.
invStdDevToVarianceWithOutput
Write the result of invStdDevToVariance() to the given output tensor,
out.See invStdDevToVariance() for the remaining parameter descriptions.
- param out
The tensor to write the variance to.
varianceToInvStdDevInPlace
Update the
variancetensor with the result of varianceToInvStdDev().See varianceToInvStdDev() for parameter descriptions.
varianceToInvStdDevWithOutput
Write the result of varianceToInvStdDev() to the given output tensor,
out.See varianceToInvStdDev() for the remaining parameter descriptions.
- param out
The tensor to write inverse standard deviation to.
-
namespace expr
Functions
- const PlaceHolder _1 (1)
- const PlaceHolder _2 (2)
- const PlaceHolder _3 (3)
- const PlaceHolder _4 (4)
- const PlaceHolder _5 (5)
- const PlaceHolder _6 (6)
- const PlaceHolder _7 (7)
- const PlaceHolder _8 (8)
- const PlaceHolder _9 (9)
- const PlaceHolder _10 (10)
- const PlaceHolder _11 (11)
- const PlaceHolder _12 (12)
- const PlaceHolder _13 (13)
- const PlaceHolder _14 (14)
- const PlaceHolder _15 (15)
- const PlaceHolder _16 (16)
- const PlaceHolder _17 (17)
- const PlaceHolder _18 (18)
- const PlaceHolder _19 (19)
- const PlaceHolder _20 (20)
-
inline BitwiseNot operator~(const Expr &a)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), Add >::type operator+ (const T &a, const Expr &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), Add >::type operator+ (const Expr &a, const T &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), BitwiseAnd >::type operator& (const T &a, const Expr &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), BitwiseAnd >::type operator& (const Expr &a, const T &b)
-
inline BitwiseAnd operator&(const Expr &a, const Expr &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), BitwiseOr >::type operator| (const T &a, const Expr &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), BitwiseOr >::type operator| (const Expr &a, const T &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), BitwiseXor >::type operator^ (const T &a, const Expr &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), BitwiseXor >::type operator^ (const Expr &a, const T &b)
-
inline BitwiseXor operator^(const Expr &a, const Expr &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), Divide >::type operator/ (const T &a, const Expr &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), Divide >::type operator/ (const Expr &a, const T &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), Equal >::type operator== (const T &a, const Expr &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), Equal >::type operator== (const Expr &a, const T &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), Gte >::type operator>= (const T &a, const Expr &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), Gte >::type operator>= (const Expr &a, const T &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), Gt >::type operator> (const T &a, const Expr &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), Gt >::type operator> (const Expr &a, const T &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), Lte >::type operator<= (const T &a, const Expr &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), Lte >::type operator<= (const Expr &a, const T &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), And >::type operator&& (const T &a, const Expr &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), And >::type operator&& (const Expr &a, const T &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), Or >::type operator|| (const T &a, const Expr &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), Or >::type operator|| (const Expr &a, const T &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), Lt >::type operator< (const T &a, const Expr &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), Lt >::type operator< (const Expr &a, const T &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), Mul >::type operator* (const T &a, const Expr &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), Mul >::type operator* (const Expr &a, const T &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), NotEqual >::type operator!= (const T &a, const Expr &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), NotEqual >::type operator!= (const Expr &a, const T &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), Rem >::type operator% (const T &a, const Expr &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), Rem >::type operator% (const Expr &a, const T &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), Shl >::type operator<< (const T &a, const Expr &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), Shl >::type operator<< (const Expr &a, const T &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), Shr >::type operator>> (const T &a, const Expr &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), Shr >::type operator>> (const Expr &a, const T &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), Sub >::type operator- (const T &a, const Expr &b)
- template<typename T> inline std::enable_if<!std::is_base_of< Expr, T >::value &&poplar::TypeTraits::isSimpleType< T >), Sub >::type operator- (const Expr &a, const T &b)
-
class Any
- #include <Expr.hpp>
A class that can contain any expression, useful for building up expression trees dynamically where the type of the outermost expression may change.
Public Functions
-
class BinaryOp : public popops::expr::ExprType<BinaryOp>
- #include <Expr.hpp>
A class to represent expressions with binary operators.
Subclassed by popops::expr::Add, popops::expr::And, popops::expr::Atan2, popops::expr::BitwiseAnd, popops::expr::BitwiseOr, popops::expr::BitwiseXnor, popops::expr::BitwiseXor, popops::expr::Divide, popops::expr::Equal, popops::expr::Gt, popops::expr::Gte, popops::expr::InvStdDevToVariance, popops::expr::Lt, popops::expr::Lte, popops::expr::Max, popops::expr::Min, popops::expr::Mul, popops::expr::NotEqual, popops::expr::Or, popops::expr::Pow, popops::expr::Rem, popops::expr::Shl, popops::expr::Shr, popops::expr::ShrSE, popops::expr::Sub, popops::expr::VarianceToInvStdDev
Public Functions
-
inline BinaryOp(BinaryOpType type, const Expr &a, const Expr &b)
-
inline BinaryOpType getOpType() const
-
inline BinaryOp(BinaryOpType type, const Expr &a, const Expr &b)
-
class Cast : public popops::expr::ExprType<Cast>
- #include <Expr.hpp>
A class to represent cast expressions.
Public Functions
-
class Const : public popops::expr::ExprType<Const>
- #include <Expr.hpp>
A class to represent constant expressions.
Subclassed by popops::expr::ConstHalf
Public Functions
-
template<typename T, typename = typename std::enable_if<poplar::TypeTraits::isSimpleType<T>(), T>::type>
inline Const(T x)
-
inline Const(poplar::TypeTraits typeTraits_, poplar::Type type_, const char *data_)
-
inline char *getData() const
-
inline const poplar::TypeTraits &getTypeTraits() const
-
double getDataAsDouble() const
-
template<typename T, typename = typename std::enable_if<poplar::TypeTraits::isSimpleType<T>(), T>::type>
-
class ConstHalf : public popops::expr::Const
- #include <Expr.hpp>
A class to represent constant expressions of type
half.
-
class Expr
- #include <Expr.hpp>
Type to represent element expressions.
This class represents an expression that can be applied to elements of tensors.
The Expr type is an abstract type which can be instantiated by its sub-classes to build up expressions, for example:
Tanh(Add(Square(_1), Const(3)))).Expressions can be applied to tensors with the popops::map() and popops::mapInPlace() functions.
Subclassed by popops::expr::ExprType< BinaryOp >, popops::expr::ExprType< Cast >, popops::expr::ExprType< Const >, popops::expr::ExprType< PlaceHolder >, popops::expr::ExprType< TernaryOp >, popops::expr::ExprType< UnaryOp >, popops::expr::ExprType< T >
Protected Types
-
using ExprClassID = void (*)(void)
Protected Functions
-
inline Expr(ExprClassID classId)
Protected Attributes
-
ExprClassID classId
-
using ExprClassID = void (*)(void)
-
template<class T>
class ExprType : public popops::expr::Expr Subclassed by popops::expr::BinaryOp, popops::expr::Cast, popops::expr::Const, popops::expr::PlaceHolder, popops::expr::TernaryOp, popops::expr::UnaryOp
Public Functions
-
inline ExprType()
Friends
- friend class Expr
-
inline ExprType()
-
class PlaceHolder : public popops::expr::ExprType<PlaceHolder>
Public Functions
-
inline PlaceHolder(unsigned index)
-
PlaceHolder(PlaceHolder&&) = default
-
PlaceHolder &operator=(PlaceHolder&&) = default
-
inline PlaceHolder(const PlaceHolder &other)
-
inline PlaceHolder &operator=(const PlaceHolder &other)
-
inline unsigned getIndex() const
Private Members
-
unsigned index
-
inline PlaceHolder(unsigned index)
-
class Select : public popops::expr::TernaryOp
- #include <Expr.hpp>
Computes the conditional ternary operation.
c ? a : b
-
class TernaryOp : public popops::expr::ExprType<TernaryOp>
- #include <Expr.hpp>
A class to represent expressions with ternary operators.
Subclassed by popops::expr::Clamp, popops::expr::Select
Public Functions
-
inline TernaryOp(TernaryOpType type, const Expr &a, const Expr &b, const Expr &c)
-
inline TernaryOpType getOpType() const
-
inline TernaryOp(TernaryOpType type, const Expr &a, const Expr &b, const Expr &c)
-
class UnaryOp : public popops::expr::ExprType<UnaryOp>
- #include <Expr.hpp>
A class to represent expressions with unary operators.
Subclassed by popops::expr::Abs, popops::expr::Asin, popops::expr::BitwiseNot, popops::expr::Cbrt, popops::expr::Ceil, popops::expr::Cos, popops::expr::Erf, popops::expr::Exp, popops::expr::Exp2, popops::expr::Expm1, popops::expr::Floor, popops::expr::GeluErf, popops::expr::Inv, popops::expr::IsFinite, popops::expr::IsInf, popops::expr::IsNaN, popops::expr::Log, popops::expr::Log1p, popops::expr::Neg, popops::expr::Not, popops::expr::Round, popops::expr::Rsqrt, popops::expr::Sigmoid, popops::expr::Signum, popops::expr::Sin, popops::expr::Sqrt, popops::expr::Square, popops::expr::Tan, popops::expr::Tanh, popops::expr::Trunc
Public Functions
-
inline UnaryOp(UnaryOpType type, const Expr &a)
-
inline UnaryOpType getOpType() const
-
inline UnaryOp(UnaryOpType type, const Expr &a)