Fix resize_bilinear type propagation
This operator supports more than just float32 outputs. PiperOrigin-RevId: 259411764
This commit is contained in:
parent
98f6e0e82b
commit
a9429e942a
@ -55,7 +55,6 @@ void SetDataTypeForAllOutputs(Model* model, Operator* op,
|
|||||||
// Do the actual output data types propagation.
|
// Do the actual output data types propagation.
|
||||||
switch (op->type) {
|
switch (op->type) {
|
||||||
case OperatorType::kDequantize:
|
case OperatorType::kDequantize:
|
||||||
case OperatorType::kResizeBilinear:
|
|
||||||
// These operators unconditionally produce float outputs
|
// These operators unconditionally produce float outputs
|
||||||
SetDataTypeForAllOutputs(model, op, ArrayDataType::kFloat);
|
SetDataTypeForAllOutputs(model, op, ArrayDataType::kFloat);
|
||||||
break;
|
break;
|
||||||
|
Loading…
Reference in New Issue
Block a user