iron.dtype Namespace Reference

Functions
	str_to_dtype (dtype_str)
	dtype_to_str (dtype)

Variables
dict	dtype_map

Detailed Description

Utilities for converting between short string names and numpy dtype objects.

Function Documentation

dtype_to_str()

iron.dtype.dtype_to_str

(

dtype

)

Convert a dtype object to its string representation.

Args:
    dtype: The dtype object to convert.

Returns:
    str: The string representation of the dtype.

str_to_dtype()

iron.dtype.str_to_dtype

(

dtype_str

)

Convert a string representation of a data type to its corresponding dtype object.

Args:
    dtype_str (str): The string representation of the data type.

Returns:
    dtype: The corresponding dtype object.

Variable Documentation

dtype_map

dict iron.dtype.dtype_map

Initial value:

=  {
    "bf16": bfloat16,
    "i8": np.int8,
    "i16": np.int16,
    "f32": np.float32,
    "i32": np.int32,
}