diff options
author | Slava Shklyaev <slavash@google.com> | 2020-10-19 14:00:05 +0100 |
---|---|---|
committer | Slava Shklyaev <slavash@google.com> | 2020-11-02 10:07:03 +0000 |
commit | e4a030ef549d8d2c279436628c961eb3f70df7d6 (patch) | |
tree | e47f78f444067e3c59fdc3439b69ed63d8340fa1 | |
parent | 895e1665d46eb542b4f57d922a114fa2aa0b7bb1 (diff) | |
download | ml-e4a030ef549d8d2c279436628c961eb3f70df7d6.tar.gz |
Update getNonExtensionSize() for TENSOR_OEM_BYTE
This makes getNonExtensionSize() behave the same as the old
nonExtensionOperandSizeOfData(). See kSizeOfDataTypeOEM.
Bug: N/A
Test: NNT_static --gtest_filter=ValidationTestModel.AddOEMOperand with change I63fa286e
Change-Id: I29496d32ac9abbeca372f36442afce7f4c8a017c
Merged-In: I29496d32ac9abbeca372f36442afce7f4c8a017c
(cherry picked from commit 79f4d47c7e34557c1a36c7a6bca16c5db541a58f)
-rw-r--r-- | nn/common/TypeUtils.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/nn/common/TypeUtils.cpp b/nn/common/TypeUtils.cpp index 15b377b87..f5bbe07f0 100644 --- a/nn/common/TypeUtils.cpp +++ b/nn/common/TypeUtils.cpp @@ -103,7 +103,6 @@ size_t getNonExtensionSize(OperandType operandType) { switch (operandType) { case OperandType::SUBGRAPH: case OperandType::OEM: - case OperandType::TENSOR_OEM_BYTE: return 0; case OperandType::TENSOR_QUANT8_ASYMM: case OperandType::BOOL: @@ -111,6 +110,7 @@ size_t getNonExtensionSize(OperandType operandType) { case OperandType::TENSOR_QUANT8_SYMM_PER_CHANNEL: case OperandType::TENSOR_QUANT8_SYMM: case OperandType::TENSOR_QUANT8_ASYMM_SIGNED: + case OperandType::TENSOR_OEM_BYTE: return 1; case OperandType::TENSOR_QUANT16_SYMM: case OperandType::TENSOR_FLOAT16: |