Package index • stablehlo

All functions

BoolAttr(): BoolAttr
BooleanType(): BooleanType
Constant(): Constant
ConstantAttr(): ConstantAttr
CustomOpBackendConfig(): CustomOpBackendConfig
DotDimensionNumbers(): DotDimensionNumbers
FloatType(): FloatType
Func(): Func
FuncBody(): FuncBody
FuncId(): FuncId
FuncInput(): FuncInput
FuncInputs(): FuncInputs
FuncOutput(): FuncOutput
FuncOutputs(): FuncOutputs
FuncValue(): FuncValue
IntegerType(): IntegerType (signed)
Op(): Op
OpInputAttr(): OpInputAttr
OpInputAttrs(): OpInputAttrs
OpInputFunc(): OpInputFunc
OpInputFuncs(): OpInputFuncs
OpInputValue(): OpInputValue
OpInputValues(): OpInputValues
OpInputs(): OpInputs
OpName(): OpName
OpOutput(): OpOutput
OpOutputs(): OpOutputs
OpSignature(): OpSignature
ScalarAttr(): ScalarAttr
Shape(): Shape
StringAttr(): StringAttr
TensorDataType: TensorDataType Base Class
TensorType(): TensorType
UnsignedType(): UnsignedType
ValueId(): ValueId
ValueType(): ValueType
ValueTypes(): ValueTypes
as_dtype(): Convert to TensorDataType
constant_attr(): Create a ConstantAttr from R values
.current_func(): Get the last function created
format_double(): Format Double Array with Scientific Notation
hlo_abs(): Abs Operator
hlo_add(): Add Operator
infer_types_after_all() hlo_after_all(): AfterAll Operator
hlo_and(): And Operator
hlo_atan2(): Atan2 Operator
infer_types_bitcast_convert() hlo_bitcast_convert(): BitcastConvert Operator
infer_types_broadcast_in_dim() hlo_broadcast_in_dim(): BroadcastInDim Operator
infer_types_case() hlo_case(): Case Operator
hlo_cbrt(): Cbrt Operator
hlo_ceil(): Ceil Operator
hlo_cholesky(): Cholesky Operator
infer_types_clamp() hlo_clamp(): Clamp Operator
hlo_closure(): Create a Closure
infer_types_compare() hlo_compare(): Compare Operator
infer_types_concatenate() hlo_concatenate(): Concatenate Operator
hlo_scalar() hlo_tensor() hlo_empty() infer_types_constant(): Create a Constant
infer_types_convert() hlo_convert(): Convert Operator
hlo_cosine(): Cosine Operator
infer_types_count_leading_zeros() hlo_count_leading_zeros(): CountLeadingZeros Operator
hlo_custom_call(): Custom Call Operation
hlo_divide(): Divide Operator
infer_types_dot_general() hlo_dot_general(): DotGeneral Operator
hlo_exponential(): Exponential Operator
hlo_exponential_minus_one(): ExponentialMinusOne Operator
hlo_floor(): Floor Operator
hlo_func() local_func(): Create a function
infer_types_if() hlo_if(): If Operator
hlo_input(): Create a input to a function
hlo_iota() infer_types_iota(): Iota Operator
infer_types_is_finite() hlo_is_finite(): IsFinite Operator
hlo_log(): Log Operator
hlo_log_plus_one(): LogPlusOne Operator
hlo_logistic(): Logistic Operator
hlo_maximum(): Maximum Operator
hlo_minimum(): Minimum Operator
hlo_multiply(): Multiply Operator
hlo_negate(): Negate Operator
hlo_not(): Not Operator
hlo_or(): Or Operator
infer_types_pad() hlo_pad(): Pad Operator
infer_types_popcnt() hlo_popcnt(): Popcnt Operator
hlo_power(): Power Operator
infer_types_reduce() hlo_reduce(): Reduce Operator
infer_types_reduce_window() hlo_reduce_window(): ReduceWindow Operator
hlo_remainder(): Remainder Operator
infer_types_reshape() hlo_reshape(): Reshape Operator
hlo_return() infer_types_return(): Return Values
infer_types_reverse() hlo_reverse(): Reverse Operator
infer_types_rng_bit_generator() hlo_rng_bit_generator(): RngBitGenerator Operator
hlo_round_nearest_afz(): RoundNearestAfz Operator
hlo_round_nearest_even(): RoundNearestEven Operator
hlo_rsqrt(): Rsqrt Operator
infer_types_select() hlo_select(): Select Operator
infer_types_shift_left() hlo_shift_left(): ShiftLeft Operator
infer_types_shift_right_arithmetic() hlo_shift_right_arithmetic(): ShiftRightArithmetic Operator
infer_types_shift_right_logical() hlo_shift_right_logical(): ShiftRightLogical Operator
hlo_sign(): Sign Operator
hlo_sine(): Sine Operator
infer_types_slice() hlo_slice(): Slice Operator
infer_types_sort() hlo_sort(): Sort Operator
hlo_sqrt(): Sqrt Operator
hlo_subtract(): Subtract Operator
hlo_tan(): Tan Operator
hlo_tanh(): Tanh Operator
infer_types_transpose() hlo_transpose(): Transpose Operator
infer_types_triangular_solve() hlo_triangular_solve(): TriangularSolve Operator
infer_types_while() hlo_while(): While Operator
hlo_xor(): Xor Operator
infer_types_custom_call(): Infer types for custom call
infer_types_generic_biv(): Infer types for binary operations
infer_types_generic_uni(): Infer types for unary operations
infer_types_integerish_biv(): Infer types for boolean integerish operations
infer_types_integerish_uni(): Infer types for integerish unary operations
is_dtype(): Is TensorDataType
r_to_constant(): Convert R value to Constant
repr(): Generate string representation for object