# SPDX-FileCopyrightText: © 2026 Tenstorrent USA, Inc.

# SPDX-License-Identifier: Apache-2.0

from models.autoports.google_embeddinggemma_2.tt.functional_encoder_layer import FunctionalEncoderLayer


class FunctionalDecoder(FunctionalEncoderLayer):
    def prefill_forward(self, hidden, per_layer_input, attn_mask, cos, sin):
        return self.forward(hidden, per_layer_input, attn_mask, cos, sin)