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

MKLDNN conv + elementwise_add fusion: bias in test marked as persistable

上级 af8c7131
...@@ -103,7 +103,7 @@ TEST(ConvElementwiseAddMKLDNNFusePass, ConvolutionWithElementwiseAddRelu) { ...@@ -103,7 +103,7 @@ TEST(ConvElementwiseAddMKLDNNFusePass, ConvolutionWithElementwiseAddRelu) {
{"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 == "weights") { if (v == "weights" || v == "bias") {
var->SetPersistable(true); var->SetPersistable(true);
} }
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册