Searched refs:BenchmarkParam (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/benchmark/ |
| D | benchmark_params.cc | 27 void BenchmarkParam::AssertHasSameType(BenchmarkParam::ParamType a, in AssertHasSameType()
28 BenchmarkParam::ParamType b) { in AssertHasSameType()
33 BenchmarkParam::ParamType BenchmarkParam::GetValueType<int32_t>() { in GetValueType()
34 return BenchmarkParam::ParamType::TYPE_INT32; in GetValueType()
38 BenchmarkParam::ParamType BenchmarkParam::GetValueType<bool>() { in GetValueType()
39 return BenchmarkParam::ParamType::TYPE_BOOL; in GetValueType()
43 BenchmarkParam::ParamType BenchmarkParam::GetValueType<float>() { in GetValueType()
44 return BenchmarkParam::ParamType::TYPE_FLOAT; in GetValueType()
48 BenchmarkParam::ParamType BenchmarkParam::GetValueType<std::string>() { in GetValueType()
49 return BenchmarkParam::ParamType::TYPE_STRING; in GetValueType()
|
| D | benchmark_test.cc | 36 params.AddParam("num_runs", BenchmarkParam::Create<int32_t>(2)); in CreateParams()
37 params.AddParam("min_secs", BenchmarkParam::Create<float>(1.0f)); in CreateParams()
38 params.AddParam("run_delay", BenchmarkParam::Create<float>(-1.0f)); in CreateParams()
39 params.AddParam("num_threads", BenchmarkParam::Create<int32_t>(1)); in CreateParams()
40 params.AddParam("benchmark_name", BenchmarkParam::Create<std::string>("")); in CreateParams()
41 params.AddParam("output_prefix", BenchmarkParam::Create<std::string>("")); in CreateParams()
42 params.AddParam("warmup_runs", BenchmarkParam::Create<int32_t>(1)); in CreateParams()
43 params.AddParam("graph", BenchmarkParam::Create<std::string>(*g_model_path)); in CreateParams()
44 params.AddParam("input_layer", BenchmarkParam::Create<std::string>("")); in CreateParams()
45 params.AddParam("input_layer_shape", BenchmarkParam::Create<std::string>("")); in CreateParams()
[all …]
|
| D | benchmark_params.h | 31 class BenchmarkParam {
39 static std::unique_ptr<BenchmarkParam> Create(const T& default_value) { in Create()
40 return std::unique_ptr<BenchmarkParam>( in Create()
49 virtual ~BenchmarkParam() {} in ~BenchmarkParam()
50 BenchmarkParam(ParamType type) : type_(type) {} in BenchmarkParam() function
59 class TypedBenchmarkParam : public BenchmarkParam {
62 : BenchmarkParam(GetValueType<T>()), value_(value) {} in TypedBenchmarkParam()
74 std::unique_ptr<BenchmarkParam> value) { in AddParam()
96 std::unordered_map<std::string, std::unique_ptr<BenchmarkParam>> params_;
|
| D | benchmark_model.cc | 44 params.AddParam("num_runs", BenchmarkParam::Create<int32_t>(50)); in DefaultParams()
45 params.AddParam("min_secs", BenchmarkParam::Create<float>(1.0f)); in DefaultParams()
46 params.AddParam("run_delay", BenchmarkParam::Create<float>(-1.0f)); in DefaultParams()
47 params.AddParam("num_threads", BenchmarkParam::Create<int32_t>(1)); in DefaultParams()
48 params.AddParam("benchmark_name", BenchmarkParam::Create<std::string>("")); in DefaultParams()
49 params.AddParam("output_prefix", BenchmarkParam::Create<std::string>("")); in DefaultParams()
50 params.AddParam("warmup_runs", BenchmarkParam::Create<int32_t>(1)); in DefaultParams()
51 params.AddParam("warmup_min_secs", BenchmarkParam::Create<float>(0.5f)); in DefaultParams()
|
| D | benchmark_tflite_model.cc | 193 default_params.AddParam("graph", BenchmarkParam::Create<std::string>("")); in DefaultParams()
195 BenchmarkParam::Create<std::string>("")); in DefaultParams()
197 BenchmarkParam::Create<std::string>("")); in DefaultParams()
198 default_params.AddParam("use_nnapi", BenchmarkParam::Create<bool>(false)); in DefaultParams()
199 default_params.AddParam("allow_fp16", BenchmarkParam::Create<bool>(false)); in DefaultParams()
|