Function main

serving/processor/tests/flatbuf_test/cpp/benchmark.cc:354–673

 
int main() {

  std::cout << "DIM_0: " << DIM_0
            << ", DIM_1: " << DIM_1
            << ", COUNT: " << COUNT << "\n";

#if DEBUG
  // if print debug log, only tesing once.
  TESTING_COUNT = 1;
#endif

  // user input
  std::string signature_name("default_serving");
  std::string input_name;
  TFDataType data_type;
  std::vector<long> shape;
  std::vector<TensorType> content;
  std::vector<std::string> string_content;
  std::vector<std::string> fetch_names;

  std::vector<std::string> input_names;
  std::vector<TFDataType> data_types;
  std::vector<std::vector<long>> shapes;
  std::vector<char*> contents;
  std::vector<std::vector<std::string>> string_contents;

#if USE_STRING_TYPE
  {
    PrepareUserStringInputs(signature_name, input_name, data_type,
                            shape, string_content, fetch_names);
  }
#else
  {
    // prepare data (will be used 10 times)
    PrepareUserInputs(signature_name, input_name, data_type,
                      shape, content, fetch_names);
  }
#endif

  for (int i = 0; i < COUNT; ++i) {
    input_names.push_back(input_name + "_" + std::to_string(i));
    data_types.push_back(data_type);
    shapes.push_back(shape);
#if USE_STRING_TYPE
    {
      string_contents.push_back(string_content);
    }
#else
    {
      contents.push_back((char*)content.data());
    }
#endif
  }

  // ------------------------ FLATBUFFER -----------------------------

  auto flat_start_timer = system_clock::now();