From c26e2f0e328d2e4a9a8d3c376623dfa45fc9b45a Mon Sep 17 00:00:00 2001 From: ANSHUMAN TRIPATHY Date: Thu, 16 May 2019 12:36:07 +0530 Subject: [PATCH] [3] Review comments handled --- tensorflow/lite/kernels/test_util.cc | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/tensorflow/lite/kernels/test_util.cc b/tensorflow/lite/kernels/test_util.cc index 1e1519befe2..b4d6eade65e 100644 --- a/tensorflow/lite/kernels/test_util.cc +++ b/tensorflow/lite/kernels/test_util.cc @@ -178,17 +178,19 @@ TfLiteStatus SingleOpModel::InvokeUnchecked() { return interpreter_->Invoke(); } void SingleOpModel::BuildInterpreter( std::vector> input_shapes) { - BuildInterpreter(input_shapes, -1, false); + BuildInterpreter(input_shapes, /*num_threads=*/-1, + /*allow_fp32_relax_to_fp16=*/false); } void SingleOpModel::BuildInterpreter(std::vector> input_shapes, int num_threads) { - BuildInterpreter(input_shapes, num_threads, false); + BuildInterpreter(input_shapes, num_threads, + /*allow_fp32_relax_to_fp16=*/false); } void SingleOpModel::BuildInterpreter(std::vector> input_shapes, bool allow_fp32_relax_to_fp16) { - BuildInterpreter(input_shapes, -1, allow_fp32_relax_to_fp16); + BuildInterpreter(input_shapes, /*num_threads=*/-1, allow_fp32_relax_to_fp16); } // static