Function main

tensorflow/lite/testing/tflite_diff_example_test.cc:20–51

#include "tensorflow/lite/testing/tflite_diff_util.h"

int main(int argc, char** argv) {
  ::tflite::InitTensorFlow();  // For Flex support.

  ::tflite::testing::DiffOptions options =
      ::tflite::testing::ParseTfliteDiffFlags(&argc, argv);
  if (options.tensorflow_model.empty()) return 1;

  int failure_count = 0;
  for (int i = 0; i < options.num_runs_per_pass; i++) {
    if (!tflite::testing::RunDiffTest(options, /*num_invocations=*/1)) {
      ++failure_count;
    }
  }
  int failures_in_first_pass = failure_count;

  if (failure_count == 0) {
    // Let's try again with num_invocations > 1 to make sure we can do multiple
    // invocations without resetting the interpreter.
    for (int i = 0; i < options.num_runs_per_pass; i++) {
      if (!tflite::testing::RunDiffTest(options, /*num_invocations=*/2)) {
        ++failure_count;
      }
    }
  }

  fprintf(stderr, "Num errors in single-inference pass: %d\n",
          failures_in_first_pass);
  fprintf(stderr, "Num errors in multi-inference pass : %d\n",
          failure_count - failures_in_first_pass);

  return failure_count != 0 ? 1 : 0;
}