tree 4e10efacf97a4977fbb807ec968881d045f69995
parent be5edc2393518f571e4f1073f1aa3f38f559f1f1
author Haoliang Zhang <haoliang@google.com> 1639525982 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> 1639526756 -0800

In InterpreterBuilder, set number of threads after all subgraphs are added.

*Previously, in InterpreterBuilder::operator()(std::unique_ptr<Interpreter>* interpreter) method, we call interpreter->SetNumThreads(num_threads_) before calling (*interpreter)->AddSubgraphs. As a result, the thread number will only be set to the TfLiteContext of the main subgraph. After the fix, the thread number will be set to each subgraph's TfLiteContext.
*This CL also changes the benchmark tflite tool to use InterpreterBuilder(std::unique_ptr<Interpreter>* interpreter) API instead of the deprecated one.

PiperOrigin-RevId: 416414642
Change-Id: I96f1ec74684adc1711bafc8e268ec42b4abaebf6