| # 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) | |