Method encode

exir/tests/transformer.py:97–108

(self, src_tokens)

        ]

    def encode(self, src_tokens):
        # embed = self.token_embed_table(src_tokens) * self.embed_scale # fail in runtime because of lacking broadcasting
        embed = self.token_embed_table(src_tokens)
        # TODO: add the support for positional embedding

        # BxTxC -> TxBxC
        x = embed.transpose(0, 1)

        for layer in self.encoder_layers:
            x = layer(x)

        return x

    def get_random_inputs(self, method):
        if method == "encode":