Package index
-
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
-
GatherDimensionNumbers() - GatherDimensionNumbers
-
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
-
ScatterDimensionNumbers() - ScatterDimensionNumbers
-
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
-
error_concatenate_shapes() - ErrorConcatenateShapes
-
error_dim_size_mismatch() - ErrorDimSizeMismatch
-
error_dimension_uniqueness() - ErrorDimensionUniqueness
-
error_index_in_set() - ErrorIndexInSet
-
error_index_out_of_bounds() - ErrorIndexOutOfBounds
-
error_indices_not_sorted() - ErrorIndicesNotSorted
-
error_permute_index() - ErrorPermuteIndex
-
error_stablehlo() - ErrorStablehlo
-
error_unequal_types() - ErrorUnequalTypes
-
error_unexpected_list_type() - ErrorUnexpectedListType
-
format_double() - Format Double Array with Scientific Notation
-
infer_types_abs()hlo_abs() - Abs Operator
-
infer_types_add()hlo_add() - Add Operator
-
infer_types_after_all()hlo_after_all() - AfterAll Operator
-
infer_types_and()hlo_and() - And Operator
-
infer_types_atan2()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
-
infer_types_cbrt()hlo_cbrt() - Cbrt Operator
-
infer_types_ceil()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
-
infer_types_cosine()hlo_cosine() - Cosine Operator
-
infer_types_count_leading_zeros()hlo_count_leading_zeros() - CountLeadingZeros Operator
-
hlo_custom_call() - Custom Call Operation
-
infer_types_divide()hlo_divide() - Divide Operator
-
infer_types_dot_general()hlo_dot_general() - DotGeneral Operator
-
infer_types_dynamic_slice()hlo_dynamic_slice() - DynamicSlice Operator
-
infer_types_dynamic_update_slice()hlo_dynamic_update_slice() - DynamicUpdateSlice Operator
-
infer_types_exponential()hlo_exponential() - Exponential Operator
-
infer_types_exponential_minus_one()hlo_exponential_minus_one() - ExponentialMinusOne Operator
-
infer_types_floor()hlo_floor() - Floor Operator
-
hlo_func()local_func() - Create a function
-
infer_types_gather()hlo_gather() - Gather Operator
-
infer_types_if()hlo_if() - If Operator
-
hlo_input() - Create a input to a function
-
infer_types_iota()hlo_iota() - Iota Operator
-
infer_types_is_finite()hlo_is_finite() - IsFinite Operator
-
infer_types_log()hlo_log() - Log Operator
-
infer_types_log_plus_one()hlo_log_plus_one() - LogPlusOne Operator
-
infer_types_logistic()hlo_logistic() - Logistic Operator
-
infer_types_maximum()hlo_maximum() - Maximum Operator
-
infer_types_minimum()hlo_minimum() - Minimum Operator
-
infer_types_multiply()hlo_multiply() - Multiply Operator
-
infer_types_negate()hlo_negate() - Negate Operator
-
infer_types_not()hlo_not() - Not Operator
-
infer_types_or()hlo_or() - Or Operator
-
infer_types_pad()hlo_pad() - Pad Operator
-
infer_types_popcnt()hlo_popcnt() - Popcnt Operator
-
infer_types_power()hlo_power() - Power Operator
-
infer_types_reduce()hlo_reduce() - Reduce Operator
-
infer_types_reduce_window()hlo_reduce_window() - ReduceWindow Operator
-
infer_types_remainder()hlo_remainder() - Remainder Operator
-
infer_types_reshape()hlo_reshape() - Reshape Operator
-
infer_types_return()hlo_return() - Return Values
-
infer_types_reverse()hlo_reverse() - Reverse Operator
-
infer_types_rng_bit_generator()hlo_rng_bit_generator() - RngBitGenerator Operator
-
infer_types_round_nearest_afz()hlo_round_nearest_afz() - RoundNearestAfz Operator
-
infer_types_round_nearest_even()hlo_round_nearest_even() - RoundNearestEven Operator
-
infer_types_rsqrt()hlo_rsqrt() - Rsqrt Operator
-
infer_types_scatter()hlo_scatter() - Scatter 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
-
infer_types_sign()hlo_sign() - Sign Operator
-
infer_types_sine()hlo_sine() - Sine Operator
-
infer_types_slice()hlo_slice() - Slice Operator
-
infer_types_sort()hlo_sort() - Sort Operator
-
infer_types_sqrt()hlo_sqrt() - Sqrt Operator
-
infer_types_subtract()hlo_subtract() - Subtract Operator
-
infer_types_tan()hlo_tan() - Tan Operator
-
infer_types_tanh()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
-
infer_types_xor()hlo_xor() - Xor Operator
-
index_vec() - index_vec
-
infer_types_custom_call() - Infer types for custom call
-
infer_types_float_biv() - Infer types for float binary operations
-
infer_types_float_uni() - Infer types for float unary operations
-
infer_types_generic_biv() - Infer types for binary operations
-
infer_types_generic_uni() - Infer types for unary operations
-
infer_types_integer_uni() - Infer types for integer unary operations
-
infer_types_integerish_biv() - Infer types for boolean integerish operations
-
infer_types_integerish_uni() - Infer types for integerish unary operations
-
infer_types_numeric_biv() - Infer types for numeric binary operations
-
infer_types_numeric_uni() - Infer types for numeric unary operations
-
is_dtype() - Is TensorDataType
-
r_to_constant() - Convert R value to Constant
-
repr() - Generate string representation for object
-
to_one_based() - Convert 0-based indices to 1-based