def _build(self, **kwargs): """Build the mask input layer.""" Mask = kwargs[self.name] assert tf.as_dtype(Mask.dtype).is_bool M = tf.convert_to_tensor(Mask) return M, 0.0