experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add test for combining sparsity and dynamic range quantization.

Browse Source 
PiperOrigin-RevId: 353359449
Change-Id: Icdf54473b1b54dab6abda07abe73d7b6601251d3
This commit is contained in:
Yunlu Li 2021-01-22 18:55:38 -08:00 committed by TensorFlower Gardener
parent 97afdaef07
commit 01a9c4f7c0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/mlir/lite/transforms/dense_to_sparse.cc
View File

@ -386,7 +386,7 @@ void DenseToSparse::runOnFunction() {
} else if (auto cst = dyn_cast<QConstOp>(inst)) { } else if (auto cst = dyn_cast<QConstOp>(inst)) {
auto attr = cst.value(); auto attr = cst.value();
auto type = cst.getType().cast<ShapedType>(); auto type = cst.getType().cast<ShapedType>();
std::vector<int8_t> dense_data(type.getNumElements()); std::vector<int8_t> dense_data;
dense_data.reserve(type.getNumElements()); dense_data.reserve(type.getNumElements());
for (const auto& val : attr.getValues<int8_t>()) for (const auto& val : attr.getValues<int8_t>())
dense_data.push_back(val); dense_data.push_back(val);