experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use the right input_dims[] in lite/micro/kernels/cast_test.cc

Browse Source
This commit is contained in:
rsun 2021-02-03 16:49:49 -08:00
parent 952fe2581d
commit 49fcdf28ea
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/lite/micro/kernels/cast_test.cc
View File

@ -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,