def test_mixed_float16_policy(self):
    with policy.policy_scope('mixed_float16'):
      inputs1 = keras.Input(shape=(36, 512), dtype='float16')
      inputs2 = keras.Input(shape=(36,), dtype='bool')
      average_layer = keras.layers.pooling.GlobalAveragePooling1D()
      _ = average_layer(inputs1, inputs2)