onnxruntime
cae9dab0 - allow to set a list for specifying force CPU nodes (#22431)

Commit
1 year ago
allow to set a list for specifying force CPU nodes (#22431) ### Description This change allows to set EP option "forceCpuNodeNames" in which includes a list of node names that will always be running on CPU. Usage example: ```diff opt.executionProviders[0] = { name: "webgpu", validationMode: 'wgpuOnly', storageBufferCacheMode: 'bucket', + // force node "/model/embed_tokens/Gather" to be put on CPU + forceCpuNodeNames: "/model/embed_tokens/Gather" }; ```
Author
Parents
Loading