ExprOp

#include <popops/ExprOp.hpp>

Operators used in expressions with elements of tensors.

namespace popops

Common functions, such as elementwise and reductions.

namespace expr

Unnamed Group

enum TernaryOpType

Enumeration defining operators used by Expr for building expressions.

Values:

enumerator CLAMP
enumerator SELECT
enum BinaryOpType

Values:

enumerator ADD
enumerator ATAN2
enumerator BITWISE_AND
enumerator BITWISE_OR
enumerator BITWISE_XOR
enumerator BITWISE_XNOR
enumerator DIVIDE
enumerator EQUAL
enumerator GREATER_THAN_EQUAL
enumerator GREATER_THAN
enumerator INV_STD_DEV_TO_VARIANCE
enumerator LESS_THAN_EQUAL
enumerator LOGICAL_AND
enumerator LOGICAL_OR
enumerator LESS_THAN
enumerator MAXIMUM
enumerator MINIMUM
enumerator MULTIPLY
enumerator NOT_EQUAL
enumerator POWER
enumerator REMAINDER
enumerator SHIFT_LEFT
enumerator SHIFT_RIGHT
enumerator SHIFT_RIGHT_SIGN_EXTEND
enumerator SUBTRACT
enumerator VARIANCE_TO_INV_STD_DEV
enum UnaryOpType

Values:

enumerator ABSOLUTE
enumerator ASIN
enumerator BITWISE_NOT
enumerator CBRT
enumerator CEIL
enumerator COS
enumerator COUNT_LEADING_ZEROS
enumerator ERF
enumerator EXPONENT
enumerator EXPONENT_MINUS_ONE
enumerator FLOOR
enumerator GELU_ERF
enumerator INVERSE
enumerator IS_FINITE
enumerator IS_INF
enumerator IS_NAN
enumerator LOGARITHM
enumerator LOGARITHM_ONE_PLUS
enumerator LOGICAL_NOT
enumerator NEGATE
enumerator POPCOUNT
enumerator RELU
enumerator SIGNUM
enumerator SIN
enumerator TAN
enumerator TANH
enumerator ROUND
enumerator SQRT
enumerator SQUARE
enumerator SIGMOID
enumerator RSQRT
enumerator TRUNC