cerebras.modelzoo.common.half_dtype.maybe_to_half_dtype#

cerebras.modelzoo.common.half_dtype.maybe_to_half_dtype(tensor)[source]#

Return tensor cast to half dtype if on CSX or autocast CPU/GPU ctx.

previous

cerebras.modelzoo.common.half_dtype

next

cerebras.modelzoo.common.half_dtype.set_half_dtype_from_params