ParakeetTdtConvModule.load constructor

ParakeetTdtConvModule.load(
  1. Map<String, MlxArray> tensors,
  2. String prefix, {
  3. required int padding,
})

Implementation

factory ParakeetTdtConvModule.load(
  Map<String, MlxArray> tensors,
  String prefix, {
  required int padding,
}) {
  return ParakeetTdtConvModule._(
    pointwiseConv1: _Conv1dBias.load(tensors, '$prefix.pointwise_conv1'),
    depthwiseConv: _DepthwiseConv1dBias.load(
      tensors,
      '$prefix.depthwise_conv',
    ),
    batchNorm: ParakeetBatchNorm.load(tensors, '$prefix.batch_norm'),
    pointwiseConv2: _Conv1dBias.load(tensors, '$prefix.pointwise_conv2'),
    padding: padding,
  );
}