utils

package

v0.0.6 Latest Latest Go to latest Published: Oct 2, 2025 License: Apache-2.0 Imports: 7 Imported by: 0

Details

Package utils holds small utility types and functions used internally in stablehlo.

This section is empty.

This section is empty.

func DTypeToStableHLO(dtype dtypes.DType) string

func SortedKeys[T cmp.Ordered](set Set[T]) []T

SortedKeys returns the keys of the set in sorted order.

func ToSnakeCase(s string) string

ToSnakeCase converts a string from CamelCase to snake_case.

type Set[T comparable] map[T]struct{}

Set implements a Set for the key type T.

func MakeSet[T comparable](size ...int) Set[T]

MakeSet returns an empty Set of the given type. Size is optional, and if given will reserve the expected size.

func SetWith[T comparable](elements ...T) Set[T]

SetWith creates a Set[T] with the given elements inserted.

func (s Set[T]) Add(s2 Set[T]) Set[T]

Add returns the union of s and s2.

func (s Set[T]) Clone() Set[T]

Clone returns a copy of the set.

func (s Set[T]) Equal(s2 Set[T]) bool

Equal returns whether s and s2 have the exact same elements.

func (s Set[T]) Has(key T) bool

Has returns true if Set s has the given key.

func (s Set[T]) Insert(keys ...T)

Insert keys into set.

func (s Set[T]) Sub(s2 Set[T]) Set[T]

Sub returns s - s2, that is, all elements in `s` that are not in `s2`.