Public
  1. Public

onnxruntime

Public

Commits

Tianlei WuGitHub
957dccb3792
Tianlei Wu authored and GitHub committed 36c3271546b
BeamSearch op cuda (#10556)

Add BeamSearch cuda implementation with support of fp16 GPT-2 subgraph