Home
last modified time | relevance | path

Searched refs:inferenceOutputs (Results 1 – 3 of 3) sorted by relevance

/test/mlts/benchmark/jni/
Dbenchmark_jni.cpp346 jobjectArray inferenceOutputs = nullptr; in Java_com_android_nn_benchmark_core_NNTestBase_runBenchmark() local
374 inferenceOutputs = env->NewObjectArray( in Java_com_android_nn_benchmark_core_NNTestBase_runBenchmark()
375 rentry.inferenceOutputs.size(), in Java_com_android_nn_benchmark_core_NNTestBase_runBenchmark()
378 for (int i = 0;i < rentry.inferenceOutputs.size();++i) { in Java_com_android_nn_benchmark_core_NNTestBase_runBenchmark()
380 inferenceOutput = env->NewByteArray(rentry.inferenceOutputs[i].size()); in Java_com_android_nn_benchmark_core_NNTestBase_runBenchmark()
383 … memcpy(bytes, &rentry.inferenceOutputs[i][0], rentry.inferenceOutputs[i].size()); in Java_com_android_nn_benchmark_core_NNTestBase_runBenchmark()
385 env->SetObjectArrayElement(inferenceOutputs, i, inferenceOutput); in Java_com_android_nn_benchmark_core_NNTestBase_runBenchmark()
391 meanSquareErrorArray, maxSingleErrorArray, inferenceOutputs, in Java_com_android_nn_benchmark_core_NNTestBase_runBenchmark()
Drun_tflite.h57 std::vector<std::vector<uint8_t>> inferenceOutputs; member
Drun_tflite.cpp213 auto& sink = result->inferenceOutputs[output_index]; in saveInferenceOutput()
350 result.inferenceOutputs.resize(outputsCount); in benchmark()