code: def enable(dtype):
      """Set the mixed precision mode.
  
      Args:
        dtype: type to cast to.
      """
      global _mixed_precision_mode
      _mixed_precision_mode = dtype
  
file_path: sonnet/sonnet/src/mixed_precision.py
imports: import uuid
  import contextlib
  from sonnet.src import custom_getter
  from sonnet.src import utils
  import tensorflow
  import tree
label_desc: Use image node_id 1 for calling a global function with example usage: enable(dtype) without return types