diff --git a/keras_cv/src/layers/preprocessing/grid_mask.py b/keras_cv/src/layers/preprocessing/grid_mask.py index 68b303ff1a..3738e976ad 100644 --- a/keras_cv/src/layers/preprocessing/grid_mask.py +++ b/keras_cv/src/layers/preprocessing/grid_mask.py @@ -164,7 +164,7 @@ def get_random_transformation( fill_value = tf.cast(fill_value, dtype=self.compute_dtype) else: # gaussian noise - fill_value = self._random_generator.random_normal( + fill_value = self._random_generator.normal( shape=input_shape, dtype=self.compute_dtype )