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

# SPDX-License-Identifier: Apache-2.0

from models.autoports.google_embeddinggemma_2.tt.optimized_encoder_layer import OptimizedEncoderLayer


class OptimizedDecoder(OptimizedEncoderLayer):
    def prefill_forward(self, hidden, per_layer_input, attn_mask, cos, sin, plan, cu_window=None):
        return self.forward(hidden, per_layer_input, attn_mask, cos, sin, plan, cu_window)