code: def _allowed_from_types():
      return {
          dtypes.float64,
          dtypes.float32,
          dtypes.float16,
          dtypes.bfloat16,
          dtypes.half,
      }
  
file_path: tensorflow-upstream/tensorflow/compiler/tests/stochastic_cast_op_test.py
imports: import math
  from absl.testing import parameterized
  from tensorflow.compiler.tests import xla_test
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.ops import stochastic_cast_op
  from tensorflow.python.platform import test
label_desc: Use image node_id 1 for calling a global function with example usage: _allowed_from_types() and returns: unknown