Method ApplyTimestampRules

plugins/wasi_nn/MLX/model/whisper/decoding.cpp:230–236

ApplyTimestampRules implementation

// ApplyTimestampRules implementation
ApplyTimestampRules::ApplyTimestampRules(
    std::shared_ptr<whisper::Tokenizer> Tokenizer, int SampleBegin,
    std::optional<int> MaxInitialTimestampIndex)
    : Tokenizer(Tokenizer), SampleBegin(SampleBegin),
      MaxInitialTimestampIndex(MaxInitialTimestampIndex) {
  Name = "ApplyTimestampRules";
}

mx::array ApplyTimestampRules::apply(const mx::array &Logits,
                                     const mx::array &Tokens) {