diff options
| author | Kevin Cheng <kevin.cheng@arm.com> | 2021-07-28 17:19:23 -0700 |
|---|---|---|
| committer | Kevin Cheng <kevin.cheng@arm.com> | 2021-08-12 15:27:03 -0700 |
| commit | a9017401461224b9bc81e7b1c770ca6091e0e3fb (patch) | |
| tree | a9eba569a9ed3862f459b3190787cdbac57da85b /reference_model | |
| parent | 7ffccce9b9a7eeecbad0c0525f1545c2714f8556 (diff) | |
| download | reference_model-a9017401461224b9bc81e7b1c770ca6091e0e3fb.tar.gz | |
Support int4 weights read. Added conv2d int8xint4 in test generation.
Signed-off-by: Kevin Cheng <kevin.cheng@arm.com>
Change-Id: I61620f160c7dad6aac5fcc3da0a6e97f3bae5b40
Diffstat (limited to 'reference_model')
| -rw-r--r-- | reference_model/src/subgraph_traverser.cc | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/reference_model/src/subgraph_traverser.cc b/reference_model/src/subgraph_traverser.cc index d64cb38..bdf6fbc 100644 --- a/reference_model/src/subgraph_traverser.cc +++ b/reference_model/src/subgraph_traverser.cc @@ -225,7 +225,6 @@ int SubgraphTraverser::initializeGraph() DEBUG_INFO(GT, "Creating tensor %s", ts->GetName().c_str()); TosaReference::Tensor* tensor = TensorFactory::newTensor(ts->GetName(), ts->GetDtype(), ts->GetShape(), ts->GetShape().size()); - if (!ts->GetData().empty()) { if (tensor->allocate()) @@ -236,6 +235,14 @@ int SubgraphTraverser::initializeGraph() switch (ts->GetDtype()) { + case DType_INT4: + { + std::vector<int8_t> i4_data; + TosaSerializationHandler::ConvertU8toI4(ts->GetData(), tensor->getElementCount(), i4_data); + std::vector<int32_t> i32_data(i4_data.begin(), i4_data.end()); + tensor->setTensorValueInt32(i32_data.size(), i32_data.data()); + } + break; case DType_INT8: { std::vector<int8_t> i8_data; |