baker
4 years ago
1 changed files with 3 additions and 1 deletions
Unified View
Diff Options
-
+3 -1ge/offline/main.cc
+ 3
- 1
ge/offline/main.cc
View File
| @@ -282,11 +282,13 @@ class GFlagUtils { | |||||
| " --enable_compress_weight Enable compress weight. true: enable; false(default): disable\n" | " --enable_compress_weight Enable compress weight. true: enable; false(default): disable\n" | ||||
| " --compress_weight_conf Config file to compress weight\n" | " --compress_weight_conf Config file to compress weight\n" | ||||
| " --buffer_optimize Set buffer optimize. \"l2_optimize\" (default). Set \"off_optimize\" to close\n" | " --buffer_optimize Set buffer optimize. \"l2_optimize\" (default). Set \"off_optimize\" to close\n" | ||||
| " --mdl_bank_path Set the path of the custom repository generated after model tuning.\n" | |||||
| "\n[Operator Tuning]\n" | "\n[Operator Tuning]\n" | ||||
| " --precision_mode precision mode, support force_fp16(default), allow_mix_precision, " | " --precision_mode precision mode, support force_fp16(default), allow_mix_precision, " | ||||
| "allow_fp32_to_fp16, must_keep_origin_dtype.\n" | "allow_fp32_to_fp16, must_keep_origin_dtype.\n" | ||||
| " --auto_tune_mode Set tune mode. E.g.: \"GA,RL\", support configure multiple, spit by ,\n" | " --auto_tune_mode Set tune mode. E.g.: \"GA,RL\", support configure multiple, spit by ,\n" | ||||
| " --op_select_implmode Set op select implmode. Support high_precision, high_performance. " | |||||
| " --op_bank_path Set the path of the custom repository generated after operator tuning with Auto Tune.\n" | |||||
| " --op_select_implmode Set op select implmode. Support high_precision, high_performance. " | |||||
| "default: high_performance\n" | "default: high_performance\n" | ||||
| " --optypelist_for_implmode Appoint which op to select implmode, cooperated with op_select_implmode.\n" | " --optypelist_for_implmode Appoint which op to select implmode, cooperated with op_select_implmode.\n" | ||||
| " Separate multiple nodes with commas (,). Use double quotation marks (\") " | " Separate multiple nodes with commas (,). Use double quotation marks (\") " | ||||