MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / CreateParams

Function CreateParams

tensorflow/lite/tools/benchmark/benchmark_test.cc:34–60  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

32namespace {
33
34BenchmarkParams CreateParams(int32_t num_runs, float min_secs, float max_secs) {
35 BenchmarkParams params;
36 params.AddParam("num_runs", BenchmarkParam::Create<int32_t>(num_runs));
37 params.AddParam("min_secs", BenchmarkParam::Create<float>(min_secs));
38 params.AddParam("max_secs", BenchmarkParam::Create<float>(max_secs));
39 params.AddParam("run_delay", BenchmarkParam::Create<float>(-1.0f));
40 params.AddParam("num_threads", BenchmarkParam::Create<int32_t>(1));
41 params.AddParam("benchmark_name", BenchmarkParam::Create<std::string>(""));
42 params.AddParam("output_prefix", BenchmarkParam::Create<std::string>(""));
43 params.AddParam("warmup_runs", BenchmarkParam::Create<int32_t>(1));
44 params.AddParam("graph", BenchmarkParam::Create<std::string>(*g_model_path));
45 params.AddParam("input_layer", BenchmarkParam::Create<std::string>(""));
46 params.AddParam("input_layer_shape", BenchmarkParam::Create<std::string>(""));
47 params.AddParam("use_nnapi", BenchmarkParam::Create<bool>(false));
48 params.AddParam("allow_fp16", BenchmarkParam::Create<bool>(false));
49 params.AddParam("require_full_delegation",
50 BenchmarkParam::Create<bool>(false));
51 params.AddParam("warmup_min_secs", BenchmarkParam::Create<float>(0.5f));
52 params.AddParam("use_legacy_nnapi", BenchmarkParam::Create<bool>(false));
53 params.AddParam("use_gpu", BenchmarkParam::Create<bool>(false));
54 params.AddParam("enable_op_profiling", BenchmarkParam::Create<bool>(false));
55 params.AddParam("max_profiling_buffer_entries",
56 BenchmarkParam::Create<int32_t>(1024));
57 params.AddParam("nnapi_accelerator_name",
58 BenchmarkParam::Create<std::string>(""));
59 return params;
60}
61
62BenchmarkParams CreateParams() { return CreateParams(2, 1.0f, 150.0f); }
63

Callers 3

TESTFunction · 0.85
InitializeMethod · 0.85
GetNextInternalMethod · 0.85

Calls 1

AddParamMethod · 0.45

Tested by

no test coverage detected