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

internal test fix

Browse Source 
PiperOrigin-RevId: 261610779
This commit is contained in:
A. Unique TensorFlower 2019-08-04 20:24:00 -07:00 committed by TensorFlower Gardener
parent df79f08595
commit 25830f35f4
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/lite/experimental/micro/kernels/arg_min_max_test.cc
View File

@ -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 = {