diff options
Diffstat (limited to 'tensorflow/contrib/lite/toco/toco_cmdline_flags.cc')
-rw-r--r-- | tensorflow/contrib/lite/toco/toco_cmdline_flags.cc | 24 |
1 files changed, 12 insertions, 12 deletions
diff --git a/tensorflow/contrib/lite/toco/toco_cmdline_flags.cc b/tensorflow/contrib/lite/toco/toco_cmdline_flags.cc index b6aebc0470..cff79776bc 100644 --- a/tensorflow/contrib/lite/toco/toco_cmdline_flags.cc +++ b/tensorflow/contrib/lite/toco/toco_cmdline_flags.cc @@ -167,11 +167,11 @@ bool ParseTocoFlagsFromCommandLineFlags( "converted float model. Model size will be reduced and there will " "be latency improvements (at the cost of accuracy)."), // WARNING: Experimental interface, subject to change - Flag("allow_eager_ops", parsed_flags.allow_eager_ops.bind(), - parsed_flags.allow_eager_ops.default_value(), ""), + Flag("allow_flex_ops", parsed_flags.allow_flex_ops.bind(), + parsed_flags.allow_flex_ops.default_value(), ""), // WARNING: Experimental interface, subject to change - Flag("force_eager_ops", parsed_flags.force_eager_ops.bind(), - parsed_flags.force_eager_ops.default_value(), "")}; + Flag("force_flex_ops", parsed_flags.force_flex_ops.bind(), + parsed_flags.force_flex_ops.default_value(), "")}; bool asked_for_help = *argc == 2 && (!strcmp(argv[1], "--help") || !strcmp(argv[1], "-help")); if (asked_for_help) { @@ -266,15 +266,15 @@ void ReadTocoFlagsFromCommandLineFlags(const ParsedTocoFlags& parsed_toco_flags, READ_TOCO_FLAG(split_tflite_lstm_inputs, FlagRequirement::kNone); READ_TOCO_FLAG(quantize_weights, FlagRequirement::kNone); READ_TOCO_FLAG(post_training_quantize, FlagRequirement::kNone); - READ_TOCO_FLAG(allow_eager_ops, FlagRequirement::kNone); - READ_TOCO_FLAG(force_eager_ops, FlagRequirement::kNone); + READ_TOCO_FLAG(allow_flex_ops, FlagRequirement::kNone); + READ_TOCO_FLAG(force_flex_ops, FlagRequirement::kNone); - if (parsed_toco_flags.force_eager_ops.value() && - !parsed_toco_flags.allow_eager_ops.value()) { - // TODO(ycling): Consider to enforce `allow_eager_ops` when - // `force_eager_ops` is true. - LOG(WARNING) << "--force_eager_ops should always be used with " - "--allow_eager_ops."; + if (parsed_toco_flags.force_flex_ops.value() && + !parsed_toco_flags.allow_flex_ops.value()) { + // TODO(ycling): Consider to enforce `allow_flex_ops` when + // `force_flex_ops` is true. + LOG(WARNING) << "--force_flex_ops should always be used with " + "--allow_flex_ops."; } // Deprecated flag handling. |