File size: 402 Bytes
714c70a | 1 2 3 4 5 6 7 8 9 10 11 | # 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)
|