code: def test_noisy_dense_with_policy():
      inputs = tf.convert_to_tensor(
          np.random.randint(low=0, high=7, size=(2, 2))
      )
      layer = NoisyDense(
          5, dtype="mixed_float16", name="noisy_dense_policy"
      )
      outputs = layer(inputs)
      output_signature = layer.compute_output_signature(
          tf.TensorSpec(dtype="float16", shape=(2, 2))
      )
      np.testing.assert_array_equal(
          output_signature.dtype, tf.dtypes.float16
      )
      np.testing.assert_array_equal(output_signature.shape, (2, 5))
      np.testing.assert_array_equal(outputs.dtype, "float16")
      np.testing.assert_array_equal(layer.mu_kernel.dtype, "float32")
      np.testing.assert_array_equal(layer.sigma_kernel.dtype, "float32")
  
file_path: addons/tensorflow_addons/layers/tests/noisy_dense_test.py
imports: import pytest
  import numpy
  import tensorflow
  from tensorflow import keras
  from tensorflow_addons.utils import test_utils
  from tensorflow_addons.layers.noisy_dense import NoisyDense
label_desc: Use image node_id 3 for calling a global function with example usage: test_noisy_dense_with_policy() without return types