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

address cr comments

Browse source
This commit is contained in:
Guoyu Wang 2022-02-01 11:44:34 -08:00
parent d460f5b776
commit 55e536c182
2 changed files with 4 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/core/providers/nnapi/nnapi_builtin/builders/op_builder.cc
View file

@ -2262,9 +2262,7 @@ class ResizeOpBuilder : public BaseOpBuilder {
};
/* static */ bool ResizeOpBuilder::IsQuantizedOp(const NodeUnit& node_unit) {
static bool is_quant_op_type =
GetQuantizedOpType(node_unit) == QuantizedOpType::QDQResize;
return is_quant_op_type;
return GetQuantizedOpType(node_unit) == QuantizedOpType::QDQResize;
}
void ResizeOpBuilder::AddInitializersToSkip(ModelBuilder& model_builder, const NodeUnit& node_unit) const {

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

@ -1471,9 +1471,7 @@ class ResizeOpSupportChecker : public BaseOpSupportChecker {
};
/* static */ bool ResizeOpSupportChecker::IsQuantizedOp(const NodeUnit& node_unit) {
static bool is_quant_op_type =
GetQuantizedOpType(node_unit) == QuantizedOpType::QDQResize;
return is_quant_op_type;
return GetQuantizedOpType(node_unit) == QuantizedOpType::QDQResize;
}
bool ResizeOpSupportChecker::IsOpSupportedImpl(const InitializedTensorSet& initializers, const NodeUnit& node_unit,
@ -1598,6 +1596,8 @@ bool ResizeOpSupportChecker::IsOpSupportedImpl(const InitializedTensorSet& initi
if (IsQuantizedOp(node_unit)) {
// For QDQResize, we only support uint8 output now
// TODO, add int8 support to NNAPI, and maybe move all the output type check into a virtual function
// similar to HasSupportedInputsImpl
int32_t output_type;
if (!GetType(node_unit.Outputs()[0].node_arg, output_type))
return false;