From 416047559c17b4a4acb6c66623f863d4728df722 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Wed, 20 Mar 2019 18:18:23 -0700 Subject: [PATCH] Enable TOCO transformation for Quantized TransposeConv op. PiperOrigin-RevId: 239516714 --- tensorflow/lite/toco/graph_transformations/quantize.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tensorflow/lite/toco/graph_transformations/quantize.cc b/tensorflow/lite/toco/graph_transformations/quantize.cc index cef1774d413..ee00c334336 100644 --- a/tensorflow/lite/toco/graph_transformations/quantize.cc +++ b/tensorflow/lite/toco/graph_transformations/quantize.cc @@ -67,7 +67,8 @@ bool SupportsQuantization(const Operator& op) { type == OperatorType::kRandomUniform || type == OperatorType::kResizeNearestNeighbor || 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