From 49fcdf28eace0216f5f63f42d15ca72104561ef1 Mon Sep 17 00:00:00 2001 From: rsun Date: Wed, 3 Feb 2021 16:49:49 -0800 Subject: [PATCH] Use the right input_dims[] in lite/micro/kernels/cast_test.cc --- tensorflow/lite/micro/kernels/cast_test.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/lite/micro/kernels/cast_test.cc b/tensorflow/lite/micro/kernels/cast_test.cc index 48da81dc0e4..a4e28cf1f75 100644 --- a/tensorflow/lite/micro/kernels/cast_test.cc +++ b/tensorflow/lite/micro/kernels/cast_test.cc @@ -96,7 +96,7 @@ TF_LITE_MICRO_TESTS_BEGIN TF_LITE_MICRO_TEST(CastFloatToInt8) { int8_t output_data[6]; - const int input_dims[] = {1, 3, 2}; + const int input_dims[] = {2, 3, 2}; // TODO(b/178391195): Test negative and out-of-range numbers. const float input_values[] = {100.f, 1.0f, 0.f, 0.4f, 1.999f, 1.1f}; @@ -107,7 +107,7 @@ TF_LITE_MICRO_TEST(CastFloatToInt8) { TF_LITE_MICRO_TEST(CastInt8ToFloat) { float output_data[6]; - const int input_dims[] = {1, 3, 2}; + const int input_dims[] = {2, 3, 2}; const int8_t input_values[] = {123, 0, 1, 2, 3, 4}; const float golden[] = {123.f, 0.f, 1.f, 2.f, 3.f, 4.f}; tflite::testing::TestCastInt8ToFloat(input_dims, input_values, golden,