wehub-resource-sync
95 lines
3.9 KiB
C++
95 lines
3.9 KiB
C++
/* Copyright 2020 The TensorFlow Authors. All Rights Reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
==============================================================================*/
|
|
#include "tensorflow/lite/kernels/test_delegate_providers.h"
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "tensorflow/lite/tools/command_line_flags.h"
|
|
#include "tensorflow/lite/tools/logging.h"
|
|
#include "tensorflow/lite/tools/tool_params.h"
|
|
|
|
namespace tflite {
|
|
constexpr char KernelTestDelegateProviders::kAccelerationTestConfigPath[];
|
|
constexpr char KernelTestDelegateProviders::kUseSimpleAllocator[];
|
|
constexpr char KernelTestDelegateProviders::kAllowFp16PrecisionForFp32[];
|
|
constexpr char KernelTestDelegateProviders::kDumpTFLiteModelDir[];
|
|
|
|
/*static*/ KernelTestDelegateProviders* KernelTestDelegateProviders::Get() {
|
|
static KernelTestDelegateProviders* const providers =
|
|
new KernelTestDelegateProviders();
|
|
return providers;
|
|
}
|
|
|
|
KernelTestDelegateProviders::KernelTestDelegateProviders()
|
|
: delegate_list_util_(¶ms_) {
|
|
delegate_list_util_.AddAllDelegateParams();
|
|
params_.AddParam(kAccelerationTestConfigPath,
|
|
tools::ToolParam::Create<std::string>(""));
|
|
params_.AddParam(kUseSimpleAllocator, tools::ToolParam::Create<bool>(false));
|
|
params_.AddParam(kAllowFp16PrecisionForFp32,
|
|
tools::ToolParam::Create<bool>(false));
|
|
params_.AddParam(kDumpTFLiteModelDir,
|
|
tools::ToolParam::Create<std::string>(""));
|
|
}
|
|
|
|
bool KernelTestDelegateProviders::InitFromCmdlineArgs(int* argc,
|
|
const char** argv) {
|
|
std::vector<tflite::Flag> flags = {
|
|
Flag(
|
|
kAccelerationTestConfigPath,
|
|
[this](const std::string& val, int argv_position) { // NOLINT
|
|
this->params_.Set<std::string>(kAccelerationTestConfigPath, val,
|
|
argv_position);
|
|
},
|
|
"", "Acceleration test config file for SingleOpModel",
|
|
Flag::kOptional),
|
|
Flag(
|
|
kUseSimpleAllocator,
|
|
[this](const bool& val, int argv_position) { // NOLINT
|
|
this->params_.Set<bool>(kUseSimpleAllocator, val, argv_position);
|
|
},
|
|
false, "Use Simple Memory Allocator for SingleOpModel",
|
|
Flag::kOptional),
|
|
Flag(
|
|
kAllowFp16PrecisionForFp32,
|
|
[this](const bool& val, int argv_position) { // NOLINT
|
|
this->params_.Set<bool>(kAllowFp16PrecisionForFp32, val,
|
|
argv_position);
|
|
},
|
|
false, "Compare result in fp16 precision for fp32 operations",
|
|
Flag::kOptional),
|
|
Flag(
|
|
kDumpTFLiteModelDir,
|
|
[this](const std::string& val, int argv_position) { // NOLINT
|
|
this->params_.Set<std::string>(kDumpTFLiteModelDir, val,
|
|
argv_position);
|
|
},
|
|
"", "Dump TFLite models of all run tests to target directory",
|
|
Flag::kOptional)};
|
|
delegate_list_util_.AppendCmdlineFlags(flags);
|
|
|
|
bool parse_result = tflite::Flags::Parse(argc, argv, flags);
|
|
if (!parse_result || params_.Get<bool>("help")) {
|
|
std::string usage = Flags::Usage(argv[0], flags);
|
|
TFLITE_LOG(ERROR) << usage;
|
|
// Returning false intentionally when "--help=true" is specified so that
|
|
// the caller could check the return value to decide stopping the execution.
|
|
parse_result = false;
|
|
}
|
|
return parse_result;
|
|
}
|
|
} // namespace tflite
|