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

minor update

Browse source
This commit is contained in:
Guoyu Wang 2022-02-02 17:22:55 -08:00
parent 7196cdf419
commit 7849f07109
3 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/core/providers/nnapi/nnapi_builtin/builders/helper.cc
View file

@ -124,7 +124,7 @@ bool HasValidUnaryOpQuantizedInputs(const NodeUnit& node_unit) {
return true;
}
bool HasValidBinaryOpQuantizedInputs(const NodeUnit& node_unit) {
bool HasValidBinaryOpQuantizedInputTypes(const NodeUnit& node_unit) {
auto quant_op_type = GetQuantizedOpType(node_unit);
int32_t a_input_type, b_input_type;
if (!IsQuantizedBinaryOp(quant_op_type)) {

2
onnxruntime/core/providers/nnapi/nnapi_builtin/builders/helper.h
View file

@ -112,7 +112,7 @@ bool IsQuantizedBinaryOp(QuantizedOpType quant_op_type);
// Check if a qlinear unary op has valid inputs, Qlinear[Sigmoid/AveragePool]
bool HasValidUnaryOpQuantizedInputs(const NodeUnit& node_unit);
// Check if a qlinear binary op has valid inputs, Qlinear[Conv/MatMul/Add]
bool HasValidBinaryOpQuantizedInputs(const NodeUnit& node_unit);
bool HasValidBinaryOpQuantizedInputTypes(const NodeUnit& node_unit);
// Check if the given quantized input(s) or output(s) is supported
bool IsQuantizedIOSupported(const InitializedTensorSet& initializers, const NodeUnit& node_unit,

6
onnxruntime/core/providers/nnapi/nnapi_builtin/builders/op_support_checker.cc
View file

@ -286,7 +286,7 @@ bool BinaryOpSupportChecker::HasSupportedInputOutputsImpl(const NodeUnit& node_u
if (is_qlinear_add) {
// QLinearAdd
if (!HasValidBinaryOpQuantizedInputs(node_unit))
if (!HasValidBinaryOpQuantizedInputTypes(node_unit))
return false;
}
@ -751,7 +751,7 @@ bool ConvOpSupportChecker::HasSupportedInputOutputsImpl(const NodeUnit& node_uni
return BaseOpSupportChecker::HasSupportedInputOutputsImpl(node_unit);
// QLinearConv only supports input of uint8 for now
if (!HasValidBinaryOpQuantizedInputs(node_unit))
if (!HasValidBinaryOpQuantizedInputTypes(node_unit))
return false;
return true;
@ -925,7 +925,7 @@ bool GemmOpSupportChecker::HasSupportedInputOutputsImpl(const NodeUnit& node_uni
return BaseOpSupportChecker::HasSupportedInputOutputsImpl(node_unit);
// QLinearMatMul
if (!HasValidBinaryOpQuantizedInputs(node_unit))
if (!HasValidBinaryOpQuantizedInputTypes(node_unit))
return false;
return true;