From 0e9472f57a3841b35556b5b708e13e216e30d3e3 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Mon, 14 Aug 2023 23:12:08 -0700 Subject: [PATCH] Enable mask propagation on EinsumDense layer PiperOrigin-RevId: 557026374 --- keras/layers/core/einsum_dense.py | 1 + 1 file changed, 1 insertion(+) diff --git a/keras/layers/core/einsum_dense.py b/keras/layers/core/einsum_dense.py index e1d3ca334c0..9260294e7ad 100644 --- a/keras/layers/core/einsum_dense.py +++ b/keras/layers/core/einsum_dense.py @@ -143,6 +143,7 @@ def __init__( self.bias_regularizer = regularizers.get(bias_regularizer) self.kernel_constraint = constraints.get(kernel_constraint) self.bias_constraint = constraints.get(bias_constraint) + self.supports_masking = True def build(self, input_shape): input_shape = tf.TensorShape(input_shape)