import { Tensor } from '@tensorflow/tfjs-core';
import { default as BaseLayer, ForwardAttributes } from './BaseLayer';
import { GPTConfig } from '../models/config';
export default class TiedEmbeddingOutputLayer extends BaseLayer {
    private vocabSize;
    private embedDim;
    private initializer;
    private WEIGHTS;
    constructor(config: GPTConfig, name: string, parent?: BaseLayer);
    embed(inputs: Tensor): Tensor;
    project(inputs: Tensor): Tensor;
    forward(_: ForwardAttributes, x: Tensor): Tensor;
}
