From 25830f35f470f1424e9eb0b16c023e5e01bbd1ff Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Sun, 4 Aug 2019 20:24:00 -0700 Subject: [PATCH] internal test fix PiperOrigin-RevId: 261610779 --- .../lite/experimental/micro/kernels/arg_min_max_test.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tensorflow/lite/experimental/micro/kernels/arg_min_max_test.cc b/tensorflow/lite/experimental/micro/kernels/arg_min_max_test.cc index c505f209215..28c91920667 100644 --- a/tensorflow/lite/experimental/micro/kernels/arg_min_max_test.cc +++ b/tensorflow/lite/experimental/micro/kernels/arg_min_max_test.cc @@ -317,7 +317,7 @@ TF_LITE_MICRO_TEST(GetMinArgMulDimensions) { using tflite::testing::F2Q; float input_min = 0; float input_max = 15.9375; - int32_t output_data[1]; + int32_t output_data[2]; TfLiteIntArray* input_dims = tflite::testing::IntArrayFromInitializer({4, 1, 1, 2, 4}); auto input_data = { @@ -341,7 +341,7 @@ TF_LITE_MICRO_TEST(GetMinArgOutput64) { using tflite::testing::F2Q; float input_min = 0; float input_max = 15.9375; - int64_t output_data[1]; + int64_t output_data[2]; TfLiteIntArray* input_dims = tflite::testing::IntArrayFromInitializer({4, 1, 1, 2, 4}); auto input_data = { @@ -365,7 +365,7 @@ TF_LITE_MICRO_TEST(GetMinArgAxis64) { using tflite::testing::F2Q; float input_min = 0; float input_max = 15.9375; - int32_t output_data[1]; + int32_t output_data[2]; TfLiteIntArray* input_dims = tflite::testing::IntArrayFromInitializer({4, 1, 1, 2, 4}); auto input_data = {