18#ifndef _ODLA_OPERATOR_OPS_MATH_H_
19#define _ODLA_OPERATOR_OPS_MATH_H_
__INT32_TYPE__ odla_int32
#define ODLA_API_EXPORT
API export directives.
__INT64_TYPE__ odla_int64
__UINT32_TYPE__ odla_uint32
__SIZE_TYPE__ odla_size_t
size_t
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Cos(odla_value x, const odla_value_id value_id)
Cos.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Neg(odla_value input, const odla_value_id value_id)
Flip the sign.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_ACosh(odla_value x, const odla_value_id value_id)
ACosh.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_ReduceSum(odla_value input, odla_size_t num_of_axes, const odla_uint32 *axes, odla_bool keep_dims, odla_value_shape output_dims, const odla_value_id value_id)
Compute the sum alone axes.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Mul(odla_value lhs, odla_value rhs, const odla_value_id value_id)
Multiplication.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_ReduceMin(odla_value input, odla_size_t num_of_axes, const odla_uint32 *axes, odla_bool keep_dims, odla_value_shape output_dims, const odla_value_id value_id)
Compute the min alone axes.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Or(odla_value lhs, odla_value rhs, const odla_value_id value_id)
Logic or test.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Sqrt(odla_value input, const odla_value_id value_id)
Square root.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Mod(odla_value lhs, odla_value rhs, odla_int64 fmod, const odla_value_id value_id)
Returns the element-wise modulus value.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Div(odla_value lhs, odla_value rhs, const odla_value_id value_id)
Division.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_ASinh(odla_value x, const odla_value_id value_id)
ASinh.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Greater(odla_value lhs, odla_value rhs, const odla_value_id value_id)
"Greater Than" test
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Round(odla_value input, const odla_value_id value_id)
Round to nearest.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_ReduceSumSquare(odla_value input, odla_size_t num_of_axes, const odla_uint32 *axes, odla_bool keep_dims, odla_value_shape output_dims, const odla_value_id value_id)
Compute the sum square alone axes.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_IsInf(odla_value input, odla_bool detect_pos, odla_bool detect_neg, const odla_value_id value_id)
Check infinity.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_ASin(odla_value x, const odla_value_id value_id)
ASin.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_SquaredDifference(odla_value lhs, odla_value rhs, const odla_value_id value_id)
Squared Difference.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_And(odla_value lhs, odla_value rhs, const odla_value_id value_id)
Logical and.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Exp(odla_value input, const odla_value_id value_id)
Compute exponential function.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Add(odla_value lhs, odla_value rhs, const odla_value_id value_id)
Addition.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Erf(odla_value input, const odla_value_id value_id)
Compute guass error of the given input.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Not(odla_value input, const odla_value_id value_id)
Logical negation.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_ATan(odla_value x, const odla_value_id value_id)
ATan.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Floor(odla_value input, const odla_value_id value_id)
Round down a value.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Sign(odla_value input, const odla_value_id value_id)
Sign of input.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Inverse(odla_value input, const odla_value_id value_id)
Inverse of a square matrix.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_ReduceLogSum(odla_value input, odla_size_t num_of_axes, const odla_uint32 *axes, odla_bool keep_dims, odla_value_shape output_dims, const odla_value_id value_id)
Compute the log sum alone axes.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Clamp(odla_value input, odla_float32 lo, odla_float32 hi, const odla_value_id value_id)
Clamp a value to a given range.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Ceil(odla_value input, const odla_value_id value_id)
Round up a value.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Max(odla_value lhs, odla_value rhs, const odla_value_id value_id)
Return the element-wise largest value.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_ReduceMean(odla_value input, odla_size_t num_of_axes, const odla_uint32 *axes, odla_bool keep_dims, odla_value_shape output_dims, const odla_value_id value_id)
Compute the mean alone axes.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Einsum(odla_values inputs, const odla_char *equation, odla_value_shape output_dims, const odla_value_id value_id)
Compute Einstein summation.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_BatchMatMul(odla_value A, odla_bool A_transpose, odla_value B, odla_bool B_transpose, odla_value_shape output_dims, const odla_value_id value_id)
General Batch Matrix Multiplication.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Shift(odla_value input, odla_value shift_amount, odla_bool is_left_shift, const odla_value_id value_id)
bit shift
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Sinh(odla_value x, const odla_value_id value_id)
Sinh.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Tanh(odla_value x, const odla_value_id value_id)
Tanh.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Mean(odla_values inputs, const odla_value_id value_id)
Compute the element-wise mean value of inputs.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Det(odla_value input, odla_value_shape output_shape, const odla_value_id value_id)
Compute the determinant of a square matrix.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Tan(odla_value x, const odla_value_id value_id)
Tan.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_ReduceProd(odla_value input, odla_size_t num_of_axes, const odla_uint32 *axes, odla_bool keep_dims, odla_value_shape output_dims, const odla_value_id value_id)
Compute the production alone axes.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Min(odla_value lhs, odla_value rhs, const odla_value_id value_id)
Returns the element-wise smallest value.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Pow(odla_value base, odla_value exponent, const odla_value_id value_id)
Raise to power.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_GreaterOrEqual(odla_value lhs, odla_value rhs, const odla_value_id value_id)
"Greater Than Or Equal" test
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Sin(odla_value x, const odla_value_id value_id)
Sin.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_LessOrEqual(odla_value lhs, odla_value rhs, const odla_value_id value_id)
"Less Than Or Equal" test
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Abs(odla_value input, const odla_value_id value_id)
Absolute value.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Gemm(odla_value A, odla_bool A_transpose, odla_value B, odla_bool B_transpose, odla_float32 alpha, odla_float32 beta, odla_value C, odla_value_shape output_dims, const odla_value_id value_id)
General Matrix Multiplication.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_IsNaN(odla_value input, const odla_value_id value_id)
Check whether element of input is a number.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Less(odla_value lhs, odla_value rhs, const odla_value_id value_id)
"Less Than" test
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_ReduceL1(odla_value input, odla_size_t num_of_axes, const odla_uint32 *axes, odla_bool keep_dims, odla_float32 epsilon, odla_value_shape output_dims, const odla_value_id value_id)
Compute the L1 norm alone axes.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_ACos(odla_value x, const odla_value_id value_id)
ACos.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Select(odla_value condition, odla_value a, odla_value b, odla_value_shape output_dims, const odla_value_id value_id)
conditional (ternary) operator
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Equal(odla_value lhs, odla_value rhs, const odla_value_id value_id)
Equality test.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_ATanh(odla_value x, const odla_value_id value_id)
ATanh.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_ArgMin(odla_value input, odla_int32 axis, odla_bool keep_dims, odla_bool return_last_index, odla_value_type output_value_type, const odla_value_id value_id)
Find the indices of the smallest elements.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_ReduceL2(odla_value input, odla_size_t num_of_axes, const odla_uint32 *axes, odla_bool keep_dims, odla_float32 epsilon, odla_value_shape output_dims, const odla_value_id value_id)
Compute the L2 norm alone axes.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Xor(odla_value lhs, odla_value rhs, const odla_value_id value_id)
Logic xor test.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_NotEqual(odla_value lhs, odla_value rhs, const odla_value_id value_id)
Logical or.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_ArgMax(odla_value input, odla_int32 axis, odla_bool keep_dims, odla_bool return_last_index, odla_value_type output_value_type, const odla_value_id value_id)
Find the indices of the largest elements.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Rsqrt(odla_value input, const odla_value_id value_id)
reciprocal square root
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Log(odla_value input, const odla_value_id value_id)
Compute the natural logrithm.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Reciprocal(odla_value input, const odla_value_id value_id)
Compute reciprocal.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Cosh(odla_value x, const odla_value_id value_id)
Cosh.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_Sub(odla_value lhs, odla_value rhs, const odla_value_id value_id)
Subtraction.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_ReduceMax(odla_value input, odla_size_t num_of_axes, const odla_uint32 *axes, odla_bool keep_dims, odla_value_shape output_dims, const odla_value_id value_id)
Compute the max alone axes.
ODLA_API_EXPORT odla_value ODLA_API_CALL odla_ReduceLogSumExp(odla_value input, odla_size_t num_of_axes, const odla_uint32 *axes, odla_bool keep_dims, odla_value_shape output_dims, const odla_value_id value_id)
Compute the log sum exponent alone axes.
struct _odla_value_id * odla_value_id
Unique id of each value.
struct _odla_value * odla_value
Value definition.