From 16069bf8745b029a82ee8eae194909f578b3dea0 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Wed, 12 Dec 2018 12:05:32 -0800 Subject: [PATCH] Allow RandomUniform to be quantized. As the RandomUniform operator is a custom op, it is up to the tf-lite user to provide the implementation. Best to assume it exists so the user can implement. PiperOrigin-RevId: 225228337 --- tensorflow/lite/toco/graph_transformations/quantize.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/tensorflow/lite/toco/graph_transformations/quantize.cc b/tensorflow/lite/toco/graph_transformations/quantize.cc index 1146078c301..2fa80f2edac 100644 --- a/tensorflow/lite/toco/graph_transformations/quantize.cc +++ b/tensorflow/lite/toco/graph_transformations/quantize.cc @@ -64,6 +64,7 @@ bool SupportsQuantization(const Operator& op) { type == OperatorType::kRelu1 || type == OperatorType::kRelu6 || type == OperatorType::kShape || type == OperatorType::kExpandDims || type == OperatorType::kPack || type == OperatorType::kTopK_V2 || + type == OperatorType::kRandomUniform || type == OperatorType::kResizeNearestNeighbor || type == OperatorType::kPRelu; }