From 1be46ca19bf1cdd58f4506860437879618577f44 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Ivan=20Vagin/AI=20Tools=20Lab=20/SRR/Engineer/=EC=82=BC?= =?utf8?q?=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Thu, 4 Jul 2019 05:21:17 +0300 Subject: [PATCH] [tflite_loader] Generate smaller values for input tensors (#5538) Generate smaller values for input tensors Signed-off-by: Ivan Vagin --- contrib/tflite_loader/tools/src/tflite_loader.cc | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/contrib/tflite_loader/tools/src/tflite_loader.cc b/contrib/tflite_loader/tools/src/tflite_loader.cc index 21a9311..f288fed 100644 --- a/contrib/tflite_loader/tools/src/tflite_loader.cc +++ b/contrib/tflite_loader/tools/src/tflite_loader.cc @@ -63,12 +63,12 @@ std::vector readData(const string &path) return vec; } -std::vector randomData(const size_t size) +std::vector randomData(RandomGenerator &randgen, const uint64_t size) { std::vector vec(size); - for (int i = 0; i < size; i++) + for (uint64_t i = 0; i < size; i++) { - vec[i] = (i % 256); + vec[i] = randgen.generate(); } return vec; } @@ -173,12 +173,15 @@ int main(const int argc, char **argv) << "Wrong number of input files." << std::endl; exit(1); } + + const int seed = 1; /* TODO Add an option for seed value */ + RandomGenerator randgen{seed, 0.0f, 2.0f}; for (int i = 0; i < num_inputs; i++) { if (generate_data) { - auto sz = test_graph->operands().at(test_graph->getInputs().at(i)).shape().num_elements(); - inputs[i] = randomData(sz); + uint64_t sz = test_graph->operands().at(test_graph->getInputs().at(i)).shape().num_elements(); + inputs[i] = randomData(randgen, sz); } else /* read_data */ inputs[i] = readData(data_files[i]); -- 2.7.4