jax.dtypes.canonicalize_dtype#

jax.dtypes.canonicalize_dtype(dtype, allow_extended_dtype=False, allow_opaque_dtype=None)#

Convert from a dtype to a canonical dtype based on config.x64_enabled.

Parameters:
  • dtype (Any) –

  • allow_extended_dtype (bool) –

  • allow_opaque_dtype (Any) –

Return type:

Union[dtype, ExtendedDType]