Home
last modified time | relevance | path

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

/test/mlts/benchmark/jni/
Dbenchmark_jni.cpp319 jobjectArray inferenceOutputs = nullptr; in Java_com_android_nn_benchmark_core_NNTestBase_runBenchmark() local
347 inferenceOutputs = env->NewObjectArray( in Java_com_android_nn_benchmark_core_NNTestBase_runBenchmark()
348 rentry.inferenceOutputs.size(), in Java_com_android_nn_benchmark_core_NNTestBase_runBenchmark()
351 for (int i = 0;i < rentry.inferenceOutputs.size();++i) { in Java_com_android_nn_benchmark_core_NNTestBase_runBenchmark()
353 inferenceOutput = env->NewByteArray(rentry.inferenceOutputs[i].size()); in Java_com_android_nn_benchmark_core_NNTestBase_runBenchmark()
356 … memcpy(bytes, &rentry.inferenceOutputs[i][0], rentry.inferenceOutputs[i].size()); in Java_com_android_nn_benchmark_core_NNTestBase_runBenchmark()
358 env->SetObjectArrayElement(inferenceOutputs, i, inferenceOutput); in Java_com_android_nn_benchmark_core_NNTestBase_runBenchmark()
364 meanSquareErrorArray, maxSingleErrorArray, inferenceOutputs, in Java_com_android_nn_benchmark_core_NNTestBase_runBenchmark()
Drun_tflite.h54 std::vector<std::vector<uint8_t>> inferenceOutputs; member
Drun_tflite.cpp181 auto& sink = result->inferenceOutputs[output_index]; in saveInferenceOutput()
317 result.inferenceOutputs.resize(outputsCount); in benchmark()