提交 b73b8683 编写于 作者: T Tomasz Patejko

MKLDNN conv + elementwise_add fusion: bias in tests made persistent.

test=develop
上级 a1fa2032
...@@ -204,7 +204,7 @@ TEST(ConvElementwiseAddMKLDNNFusePass, SigmoidConvolutionAddElementwiseRelu) { ...@@ -204,7 +204,7 @@ TEST(ConvElementwiseAddMKLDNNFusePass, SigmoidConvolutionAddElementwiseRelu) {
{"a", "b", "bias", "weights", "c", "d", "e", "f"})) { {"a", "b", "bias", "weights", "c", "d", "e", "f"})) {
auto* var = prog.MutableBlock(0)->Var(v); auto* var = prog.MutableBlock(0)->Var(v);
var->SetType(proto::VarType::LOD_TENSOR); var->SetType(proto::VarType::LOD_TENSOR);
if (v.find("weights")) { if (v.find("weights") || v.find("bias")) {
var->SetPersistable(true); var->SetPersistable(true);
} }
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册