File size: 437 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.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)
|