cerebras.modelzoo.common.half_dtype#

Module which provides utilities for selecting 16-bit floating point representation.

Functions

set_half_dtype_from_params

Sets the half dtype in cstorch from the given model params.