|
|
@ -261,8 +261,8 @@ struct AnalysisConfig : public NativeConfig {
|
|
|
|
|
|
|
|
|
|
|
|
void SetIncludeMode() {
|
|
|
|
void SetIncludeMode() {
|
|
|
|
ir_mode = IrPassMode::kInclude;
|
|
|
|
ir_mode = IrPassMode::kInclude;
|
|
|
|
// this pass has to be run at the beginning of all fuse passes
|
|
|
|
|
|
|
|
ir_passes = {"infer_clean_graph_pass"};
|
|
|
|
ir_passes = {"infer_clean_graph_pass"};
|
|
|
|
|
|
|
|
ir_mkldnn_passes = {"infer_clean_graph_pass"};
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// Determine whether to perform graph optimization.
|
|
|
|
// Determine whether to perform graph optimization.
|
|
|
@ -271,6 +271,8 @@ struct AnalysisConfig : public NativeConfig {
|
|
|
|
IrPassMode ir_mode{IrPassMode::kExclude};
|
|
|
|
IrPassMode ir_mode{IrPassMode::kExclude};
|
|
|
|
// passes to be excluded/included
|
|
|
|
// passes to be excluded/included
|
|
|
|
std::vector<std::string> ir_passes{"embedding_fc_lstm_fuse_pass"};
|
|
|
|
std::vector<std::string> ir_passes{"embedding_fc_lstm_fuse_pass"};
|
|
|
|
|
|
|
|
// passes to be excluded/included when MKL-DNN is enabled
|
|
|
|
|
|
|
|
std::vector<std::string> ir_mkldnn_passes{"embedding_fc_lstm_fuse_pass"};
|
|
|
|
|
|
|
|
|
|
|
|
// NOT stable yet.
|
|
|
|
// NOT stable yet.
|
|
|
|
bool use_feed_fetch_ops{true};
|
|
|
|
bool use_feed_fetch_ops{true};
|
|
|
|