提交 add63d23 编写于 作者: H hong19860320

skip to query the attr 'is_test' in batch_norm op

test=develop
上级 3797f8fa
...@@ -82,7 +82,8 @@ bool BatchNormOp::AttachImpl(const cpp::OpDesc &op_desc, lite::Scope *scope) { ...@@ -82,7 +82,8 @@ bool BatchNormOp::AttachImpl(const cpp::OpDesc &op_desc, lite::Scope *scope) {
param_.variance = param_.variance =
scope->FindVar(op_desc.Input("Variance").front())->GetMutable<Tensor>(); scope->FindVar(op_desc.Input("Variance").front())->GetMutable<Tensor>();
param_.y = scope->FindVar(op_desc.Output("Y").front())->GetMutable<Tensor>(); param_.y = scope->FindVar(op_desc.Output("Y").front())->GetMutable<Tensor>();
param_.is_test = op_desc.GetAttr<int>("is_test"); param_.is_test = true; // TODO(hong19860320) param_.is_test =
// op_desc.GetAttr<int>("is_test");
param_.use_global_stats = op_desc.GetAttr<bool>("use_global_stats"); param_.use_global_stats = op_desc.GetAttr<bool>("use_global_stats");
if (!param_.is_test) { if (!param_.is_test) {
param_.mean_out = param_.mean_out =
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册