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

Merge pull request #46906 from rsun-bdti:issue-45608-pr6

Browse Source 
PiperOrigin-RevId: 355767827
Change-Id: I61e0ae8ff9ef6cfe8f49be16771bddebed4d349a
This commit is contained in:
TensorFlower Gardener 2021-02-04 20:18:18 -08:00
commit 8db7298d14
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,