Go to the source code of this file.
|
int | main (int argc, char *argv[]) |
|
◆ main()
int main |
( |
int |
argc, |
|
|
char * |
argv[] |
|
) |
| |
Definition at line 11 of file TfLiteMobileNetQuantizedSoftmax-Armnn.cpp.
References armnn::test::ClassifierInferenceTestMain().
13 int retVal = EXIT_FAILURE;
19 std::vector<ImageSet> imageSet =
37 "mobilenet_v1_0.25_128_quant.tflite",
40 "MobilenetV1/Predictions/Reshape_1",
42 [&imageSet](
const char* dataDir,
const ModelType &) {
53 DatabaseType::DataFormat::NCHW,
58 catch (
const std::exception& e)
63 std::cerr <<
"WARNING: " << *argv <<
": An error has occurred when running " 64 "the classifier inference tests: " << e.what() << std::endl;
int ClassifierInferenceTestMain(int argc, char *argv[], const char *modelFilename, bool isModelBinary, const char *inputBindingName, const char *outputBindingName, const std::vector< unsigned int > &defaultTestCaseIds, TConstructDatabaseCallable constructDatabase, const armnn::TensorShape *inputTensorShape=nullptr)