ParakeetTdtEncoder constructor

ParakeetTdtEncoder(
  1. ParakeetTdtBundle bundle, {
  2. int? maxLayers,
  3. ParakeetEncoderTrace? onLayerComplete,
})

Implementation

ParakeetTdtEncoder(
  this.bundle, {
  this.maxLayers,
  this.onLayerComplete,
})
  : _preEncoder = ParakeetTdtPreEncoder(bundle),
    _positionalEncoding = ParakeetTdtPositionalEncoding(
      dModel: bundle.manifest.encoderHidden,
      scaleInput: bundle.manifest.xScaling,
    ),
    _blocks = List<ParakeetTdtConformerBlock>.generate(
      bundle.manifest.encoderLayers,
      (int index) => ParakeetTdtConformerBlock.load(
        bundle.tensors,
        'encoder.layers.$index',
      ),
    );