saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-18 21:21:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Register Resize op into nhwc schema for Qnn EP (#15373)

Browse source 
### Description
Register Resize op into nhwc schema for Qnn EP.

### Motivation and Context
Resize op is identified as layout sensitive op for Qnn EP, need to
register it into nhwc schema
This commit is contained in:
Hector Li 2023-04-05 08:41:16 -07:00 committed by GitHub
parent 4db10c93d1
commit a0d8dbe28d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
onnxruntime/core/graph/contrib_ops/internal_nhwc_onnx_opset.cc
View file

@ -115,6 +115,11 @@ void OpSet_Internal_NHWC_ONNX::ForEachSchema(const std::function<void(ONNX_NAMES
REGISTER_NHWC_SCHEMA(fn, SpaceToDepth, 1);
REGISTER_NHWC_SCHEMA(fn, SpaceToDepth, 13);
#if defined(USE_QNN)
REGISTER_NHWC_SCHEMA(fn, Resize, 11);
REGISTER_NHWC_SCHEMA(fn, Resize, 13);
#endif
// internal QLinear ops
REGISTER_NHWC_SCHEMA_FROM_MSDOMAIN(fn, QLinearAveragePool, 1);
REGISTER_NHWC_SCHEMA_FROM_MSDOMAIN(fn, QLinearConvTranspose, 1);