From abdeb0e08d6b4964c63f64656f7917fa199b9a7f Mon Sep 17 00:00:00 2001 From: MyPandaShaoxiang Date: Sun, 2 Feb 2020 21:04:13 -0500 Subject: [PATCH] transformer_patch --- lite/core/mir/variable_place_inference_pass.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lite/core/mir/variable_place_inference_pass.h b/lite/core/mir/variable_place_inference_pass.h index 875bf23082..f243a14949 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()); } } } -- GitLab