message: Accepts key `num_dims` which is defined in message RnnState of model_flags.proto.
diff: @@ -22,11 +22,11 @@ limitations under the License.
  #include "absl/strings/str_split.h"
  #include "absl/strings/string_view.h"
  #include "absl/strings/strip.h"
  +#include "tensorflow/core/platform/logging.h"
  +#include "tensorflow/core/util/command_line_flags.h"
  #include "tensorflow/lite/toco/args.h"
  #include "tensorflow/lite/toco/toco_graphviz_dump_options.h"
  #include "tensorflow/lite/toco/toco_port.h"
  -#include "tensorflow/core/platform/logging.h"
  -#include "tensorflow/core/util/command_line_flags.h"
  // "batch" flag only exists internally
  #ifdef PLATFORM_GOOGLE
  @@ -367,6 +367,11 @@ void ReadModelFlagsFromCommandLineFlags(
  CHECK(absl::SimpleAtoi(value, &size));
  CHECK_GT(size, 0);
  rnn_state_proto->set_size(size);
  + } else if (key == "num_dims") {
  + int32 size = 0;
  + CHECK(absl::SimpleAtoi(value, &size));
  + CHECK_GT(size, 0);
  + rnn_state_proto->set_num_dims(size);
  } else {
  LOG(FATAL) << "Unknown key '" << key << "' in --rnn_states";
  }
  