Home
last modified time | relevance | path

Searched refs:ToolParam (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/
Dtool_params.cc27 void ToolParam::AssertHasSameType(ToolParam::ParamType a, in AssertHasSameType()
28 ToolParam::ParamType b) { in AssertHasSameType()
33 ToolParam::ParamType ToolParam::GetValueType<int32_t>() { in GetValueType()
34 return ToolParam::ParamType::TYPE_INT32; in GetValueType()
38 ToolParam::ParamType ToolParam::GetValueType<bool>() { in GetValueType()
39 return ToolParam::ParamType::TYPE_BOOL; in GetValueType()
43 ToolParam::ParamType ToolParam::GetValueType<float>() { in GetValueType()
44 return ToolParam::ParamType::TYPE_FLOAT; in GetValueType()
48 ToolParam::ParamType ToolParam::GetValueType<std::string>() { in GetValueType()
49 return ToolParam::ParamType::TYPE_STRING; in GetValueType()
[all …]
Dtool_params_test.cc27 params.AddParam("some-int1", ToolParam::Create<int>(13)); in TEST()
28 params.AddParam("some-int2", ToolParam::Create<int>(17)); in TEST()
31 others.AddParam("some-int1", ToolParam::Create<int>(19)); in TEST()
32 others.AddParam("some-bool", ToolParam::Create<bool>(true)); in TEST()
46 params.AddParam("some-int1", ToolParam::Create<int>(13)); in TEST()
47 params.AddParam("some-int2", ToolParam::Create<int>(17)); in TEST()
50 others.AddParam("some-int1", ToolParam::Create<int>(19)); in TEST()
51 others.AddParam("some-bool", ToolParam::Create<bool>(true)); in TEST()
61 params.AddParam("some-int1", ToolParam::Create<int>(13)); in TEST()
62 params.AddParam("some-int2", ToolParam::Create<int>(17)); in TEST()
[all …]
Dtool_params.h30 class ToolParam {
38 static std::unique_ptr<ToolParam> Create(const T& default_value) { in Create()
39 return std::unique_ptr<ToolParam>(new TypedToolParam<T>(default_value)); in Create()
54 virtual ~ToolParam() {} in ~ToolParam()
55 explicit ToolParam(ParamType type) : has_value_set_(false), type_(type) {} in ToolParam() function
59 virtual void Set(const ToolParam&) {} in Set() argument
61 virtual std::unique_ptr<ToolParam> Clone() const = 0;
73 class TypedToolParam : public ToolParam {
76 : ToolParam(GetValueType<T>()), value_(value) {} in TypedToolParam()
85 void Set(const ToolParam& other) override { in Set()
[all …]
/external/tensorflow/tensorflow/lite/tools/delegates/
Dnnapi_delegate_provider.cc28 default_params_.AddParam("use_nnapi", ToolParam::Create<bool>(false)); in NnapiDelegateProvider()
30 ToolParam::Create<std::string>("")); in NnapiDelegateProvider()
32 ToolParam::Create<std::string>("")); in NnapiDelegateProvider()
34 ToolParam::Create<std::string>("")); in NnapiDelegateProvider()
36 ToolParam::Create<bool>(true)); in NnapiDelegateProvider()
38 ToolParam::Create<bool>(false)); in NnapiDelegateProvider()
Dgpu_delegate_provider.cc36 default_params_.AddParam("use_gpu", ToolParam::Create<bool>(false)); in GpuDelegateProvider()
39 ToolParam::Create<bool>(true)); in GpuDelegateProvider()
41 ToolParam::Create<bool>(true)); in GpuDelegateProvider()
44 default_params_.AddParam("gpu_backend", ToolParam::Create<std::string>("")); in GpuDelegateProvider()
48 ToolParam::Create<std::string>("")); in GpuDelegateProvider()
Ddefault_execution_provider.cc28 default_params_.AddParam("num_threads", ToolParam::Create<int32_t>(1)); in DefaultExecutionProvider()
30 ToolParam::Create<int32_t>(0)); in DefaultExecutionProvider()
32 ToolParam::Create<int32_t>(0)); in DefaultExecutionProvider()
Dhexagon_delegate_provider.cc36 default_params_.AddParam("use_hexagon", ToolParam::Create<bool>(false)); in HexagonDelegateProvider()
38 ToolParam::Create<std::string>("/data/local/tmp")); in HexagonDelegateProvider()
40 ToolParam::Create<bool>(false)); in HexagonDelegateProvider()
Dcoreml_delegate_provider.cc35 default_params_.AddParam("use_coreml", ToolParam::Create<bool>(false)); in CoreMlDelegateProvider()
36 default_params_.AddParam("coreml_version", ToolParam::Create<int>(0)); in CoreMlDelegateProvider()
Dexternal_delegate_provider.cc43 ToolParam::Create<std::string>("")); in ExternalDelegateProvider()
45 ToolParam::Create<std::string>("")); in ExternalDelegateProvider()
Dxnnpack_delegate_provider.cc26 default_params_.AddParam("use_xnnpack", ToolParam::Create<bool>(false)); in XnnpackDelegateProvider()
/external/tensorflow/tensorflow/lite/tools/benchmark/
Dbenchmark_params.h22 using BenchmarkParam = tflite::tools::ToolParam;
/external/tensorflow/tensorflow/lite/delegates/utils/dummy_delegate/
Ddummy_delegate_provider.cc27 ToolParam::Create<bool>(false)); in DummyDelegateProvider()
/external/tensorflow/tensorflow/lite/g3doc/performance/
Dimplementing_delegate.md314 ToolParam::Create<bool>(false));