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

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

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