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

Update type reduction operator type usage processors set. (#6976)

Browse source
This commit is contained in:
Edward Chen 2021-03-11 09:22:53 -08:00 committed by GitHub
parent b57a85d863
commit aa60a8368f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
tools/python/util/ort_format_model/operator_type_usage_processors.py
View file

@ -314,9 +314,10 @@ def _create_operator_type_usage_processors():
#
# Ops we are ignoring currently so as not to produce meaningless/unused output:
# - Implementation is type agnostic:
# ai.onnx: If, Loop, Reshape, Scan, Shape, Squeeze, Unsqueeze
# ai.onnx: If, Loop, Reshape, Scan, Shape, Squeeze, Tile, Unsqueeze
# com.microsoft: DynamicQuantizeMatMul, MatMulIntegerToFloat
# - Only one type supported in the ORT implementation:
# ai.onnx: NonMaxSuppression
# com.microsoft: FusedConv, FusedGemm, FusedMatMul
# - Implementation does not have any significant type specific code:
# ai.onnx: Concat, Flatten, Not, QLinearConv, Reshape, Shape, Squeeze, Unsqueeze
@ -330,14 +331,15 @@ def _create_operator_type_usage_processors():
'Gemm',
'IsNaN',
'Log', 'LogSoftmax', 'LpNormalization',
'MatMul', 'Max', 'Min',
'Neg', 'NonMaxSuppression', 'NonZero',
'MatMul', 'Max', 'MaxPool', 'Mean', 'Min',
'Neg', 'NonZero',
'Pad',
'Range', 'Reciprocal', 'ReduceL1', 'ReduceL2', 'ReduceLogSum', 'ReduceLogSumExp',
'ReduceMax', 'ReduceMean', 'ReduceMin', 'ReduceProd', 'ReduceSum', 'ReduceSumSquare',
'Relu', 'Resize', 'RoiAlign', 'Round',
'Sigmoid', 'Sin', 'Softmax', 'Split', 'Sqrt',
'Tanh', 'Tile', 'TopK', 'Transpose',
'Relu', 'Resize', 'ReverseSequence', 'RoiAlign', 'Round',
'ScatterND', 'Sigmoid', 'Sin', 'Softmax', 'Split', 'Sqrt', 'Sum',
'Tanh', 'TopK', 'Transpose',
'Unique',
'Where']
default_processor_onnx_ops_requiring_int64_for_input_0 = ['Add',