code: def test_with_mixed_precision_policy():
      policy = tf.keras.mixed_precision.global_policy()
  
      shortcut = np.asarray([[0.2, 0.1, 0.4]])
      residual = np.asarray([[0.2, 0.4, 0.5]])
  
      output = StochasticDepth()([shortcut, residual])
      assert output.dtype == policy.compute_dtype
  
      output = StochasticDepth()([shortcut, residual], training=True)
      assert output.dtype == policy.compute_dtype
  
file_path: addons/tensorflow_addons/layers/tests/stochastic_depth_test.py
imports: import pytest
  import numpy
  import tensorflow
  from tensorflow_addons.layers.stochastic_depth import StochasticDepth
  from tensorflow_addons.utils import test_utils
label_desc: Use image node_id 2 for calling a global function with example usage: test_with_mixed_precision_policy() without return types