mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-31 23:27:43 +00:00
1de79d3158
- Add 64-bit for those which now support true 64-bit (Gather, Scatter, OneHot, Cast), and update others (ArgMin, ArgMax, ReverseSequence, TopK, MaxPool, MaxUnpool) which take 64-bit indices but use strided 32-bit fallback. - Stop forcibly coercing all 64-bit tensors in TensorDesc. Instead, decide in the respective kernels how to behave. - Update graph partitioning code with enough registration information to know whether (a) 64-bit tenors are not supported at all (b) they are support via strided 32-bit fallback (c) they are supported via fallback and directly (preferred when device capable). Unfortunately this introduces a lot of flag parameters :/. Related work items: #22265955 |
||
|---|---|---|
| .. | ||
| adapter | ||
| api | ||
| dll | ||
| lib | ||
| test | ||