diff --git a/lite/core/mir/variable_place_inference_pass.h b/lite/core/mir/variable_place_inference_pass.h index 875bf23082a24cb6fcae878b46cc9dcdbb2b76f7..f243a149494493bbedb96485385de8018e8f8fd1 100644 --- a/lite/core/mir/variable_place_inference_pass.h +++ b/lite/core/mir/variable_place_inference_pass.h @@ -142,7 +142,7 @@ class VariablePlaceInferencePass : public DebugPass { } else { PrecisionType tmp_ptype = x_in->AsArg().type->precision(); x_in->AsArg().type = LiteType::GetTensorTy( - type->target(), tmp_ptype, type->layout()); + type->target(), type->precision(), type->layout()); } } } @@ -173,7 +173,7 @@ class VariablePlaceInferencePass : public DebugPass { } else { PrecisionType tmp_ptype = x_out->AsArg().type->precision(); x_out->AsArg().type = LiteType::GetTensorTy( - type->target(), tmp_ptype, type->layout()); + type->target(), type->precision(), type->layout()); } } }