Commit 60037b67 authored by Haoliang Zhang's avatar Haoliang Zhang Committed by TensorFlower Gardener
Browse files

Use cached input to speed up tensor initialization in tflite benchmark model.

This CL changes how data inputs are generated for benchmark_tflite_model:
Previous: We random initialize all the input during each run, which is time consuming. And also contribute to the mysterious 'other' tag in the benchmark result.
After: We will initialize input only once before inference, and in each run reset the input from the cache.

After this change each run will have exactly the same data input, which shouldn't be a big concern (especially for larger models).

PiperOrigin-RevId: 236875318
parent 46ebf026
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment