File tree Expand file tree Collapse file tree 2 files changed +4
-1
lines changed Expand file tree Collapse file tree 2 files changed +4
-1
lines changed Original file line number Diff line number Diff line change @@ -58,7 +58,7 @@ std::unique_ptr<ir::Graph> ConvReLUFusePass::ApplyImpl(
58
58
desc.SetInput (" Input" , std::vector<std::string>({conv_relu_i_in}));
59
59
desc.SetInput (" Filter" , std::vector<std::string>({conv_relu_w_in}));
60
60
desc.SetInput (" Bias" , std::vector<std::string>({conv_relu_b_in}));
61
- desc.SetOutput (" Out " , std::vector<std::string>({conv_relu_out}));
61
+ desc.SetOutput (" Output " , std::vector<std::string>({conv_relu_out}));
62
62
desc.SetType (" conv2d" );
63
63
for (auto & attr : conv->Op ()->GetAttrMap ()) {
64
64
desc.SetAttr (attr.first , attr.second );
Original file line number Diff line number Diff line change @@ -72,6 +72,9 @@ class Analyzer : public OrderedRegistry<PassManager> {
72
72
" mul_gru_fuse_pass" , //
73
73
" seq_concat_fc_fuse_pass" , //
74
74
" fc_fuse_pass" , //
75
+ #ifdef PADDLE_WITH_MKLDNN
76
+ " conv_relu_mkldnn_fuse_pass" , //
77
+ #endif
75
78
}};
76
79
77
80
std::unordered_set<std::string> disabled_ir_passes_;
You can’t perform that action at this time.
0 commit comments