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
This commit is contained in:
A. Unique TensorFlower 2018-12-12 12:05:32 -08:00 committed by TensorFlower Gardener
parent 350791003d
commit 16069bf874

View File

@ -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;
}