Skip to content

P
Paddle-Lite

PaddlePaddle / Paddle-Lite

通知 331
Star 4
Fork 1
P
Paddle-Lite
提交 abdeb0e0 编写于 作者: M MyPandaShaoxiang
浏览文件
操作

transformer_patch

上级 d035f1ba
隐藏空白更改
内联 并排
Showing with 2 addition and 2 deletion
lite/core/mir/variable_place_inference_pass.h
浏览文件 @ abdeb0e0
...@@ -142,7 +142,7 @@ class VariablePlaceInferencePass : public DebugPass { ...@@ -142,7 +142,7 @@ class VariablePlaceInferencePass : public DebugPass {
} else { } else {
PrecisionType tmp_ptype = x_in->AsArg().type->precision(); PrecisionType tmp_ptype = x_in->AsArg().type->precision();
x_in->AsArg().type = LiteType::GetTensorTy( 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 { ...@@ -173,7 +173,7 @@ class VariablePlaceInferencePass : public DebugPass {
} else { } else {
PrecisionType tmp_ptype = x_out->AsArg().type->precision(); PrecisionType tmp_ptype = x_out->AsArg().type->precision();
x_out->AsArg().type = LiteType::GetTensorTy( x_out->AsArg().type = LiteType::GetTensorTy(
type->target(), tmp_ptype, type->layout()); type->target(), type->precision(), type->layout());
} }
} }
} }
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或