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

Enable TOCO transformation for Quantized TransposeConv op.

Browse Source 
PiperOrigin-RevId: 239516714
This commit is contained in:
A. Unique TensorFlower 2019-03-20 18:18:23 -07:00 committed by TensorFlower Gardener
parent aee0b43932
commit 416047559c
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/lite/toco/graph_transformations/quantize.cc
View File

@ -67,7 +67,8 @@ bool SupportsQuantization(const Operator& op) {
type == OperatorType::kRandomUniform || type == OperatorType::kRandomUniform ||
type == OperatorType::kResizeNearestNeighbor || type == OperatorType::kResizeNearestNeighbor ||
type == OperatorType::kPRelu || type == OperatorType::kReduceMax || type == OperatorType::kPRelu || type == OperatorType::kReduceMax ||
type == OperatorType::kReduceMin; type == OperatorType::kReduceMin ||
type == OperatorType::kTransposeConv;
} }
// The quantized op allows output arrays of type float using // The quantized op allows output arrays of type float using