Fix resize_bilinear type propagation

This operator supports more than just float32 outputs.

PiperOrigin-RevId: 259411764
This commit is contained in:
Jared Duke 2019-07-22 14:51:30 -07:00 committed by TensorFlower Gardener
parent 98f6e0e82b
commit a9429e942a

View File

@ -55,7 +55,6 @@ void SetDataTypeForAllOutputs(Model* model, Operator* op,
// Do the actual output data types propagation.
switch (op->type) {
case OperatorType::kDequantize:
case OperatorType::kResizeBilinear:
// These operators unconditionally produce float outputs
SetDataTypeForAllOutputs(model, op, ArrayDataType::kFloat);
break;