parameters method

  1. @override
List<Value> parameters()
override

Implementation

@override
List<Value> parameters() {
  final params = <Value>[];
  if (frameProjection != null) {
    params.addAll(frameProjection!.parameters());
  }
  params.addAll(positionEmbeddings.expand((vec) => vec.values));
  params.addAll(transformerEncoder.parameters());
  params.addAll(mlpHead.parameters());
  return params;
}