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

Merge pull request #47304 from stephanboner:PR6

Browse Source 
PiperOrigin-RevId: 360959235
Change-Id: Ib09e1206e0558f7b9f8e5c99d6aab2801f89ce1e
This commit is contained in:
TensorFlower Gardener 2021-03-04 11:18:43 -08:00
commit 3e19502621
4 changed files with 0 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -51,10 +51,6 @@ TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {
TF_LITE_ENSURE(context, NumDimensions(output) <= kInputOutputMaxDimensionNum); TF_LITE_ENSURE(context, NumDimensions(output) <= kInputOutputMaxDimensionNum);
TF_LITE_ENSURE_TYPES_EQ(context, input->type, output->type); TF_LITE_ENSURE_TYPES_EQ(context, input->type, output->type);
// Padding can result in a larger output than input.
TF_LITE_ENSURE(context,
ElementCount(*output->dims) >= ElementCount(*input->dims));
return kTfLiteOk; return kTfLiteOk;
} }

14
tensorflow/lite/micro/kernels/batch_to_space_nd_test.cc
View File

@ -151,18 +151,4 @@ TF_LITE_MICRO_TEST(BatchToSpaceBasicInt8) {
tflite::testing::basic_golden, golden_quantized, 1.0f, 0, output)); tflite::testing::basic_golden, golden_quantized, 1.0f, 0, output));
} }
TF_LITE_MICRO_TEST(BatchToSpaceInvalidOutputDimensionShouldFail) {
constexpr int output_length = 12;
const int output_dims[] = {4, 1, 4, 3, 1};
float output[output_length];
TF_LITE_MICRO_EXPECT_EQ(
kTfLiteError,
tflite::testing::TestBatchToSpaceNdFloat(
tflite::testing::basic_input_dims, tflite::testing::basic_input,
tflite::testing::basic_block_shape_dims,
tflite::testing::basic_block_shape, tflite::testing::basic_crops_dims,
tflite::testing::basic_crops, output_dims,
tflite::testing::basic_golden, output));
}
TF_LITE_MICRO_TESTS_END TF_LITE_MICRO_TESTS_END

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

@ -57,10 +57,6 @@ TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {
TF_LITE_ENSURE(context, NumDimensions(output) <= kInputOutputMaxDimensionNum); TF_LITE_ENSURE(context, NumDimensions(output) <= kInputOutputMaxDimensionNum);
TF_LITE_ENSURE_TYPES_EQ(context, input->type, output->type); TF_LITE_ENSURE_TYPES_EQ(context, input->type, output->type);
// Padding can result in a larger output than input.
TF_LITE_ENSURE(context,
ElementCount(*output->dims) >= ElementCount(*input->dims));
return kTfLiteOk; return kTfLiteOk;
} }

14
tensorflow/lite/micro/kernels/space_to_batch_nd_test.cc
View File

@ -151,18 +151,4 @@ TF_LITE_MICRO_TEST(SpaceToBatchBasicInt8) {
tflite::testing::basic_golden, golden_quantized, 1.0f, 0, output)); tflite::testing::basic_golden, golden_quantized, 1.0f, 0, output));
} }
TF_LITE_MICRO_TEST(SpaceToBatchInvalidOutputDimensionShouldFail) {
constexpr int output_length = 12;
const int output_dims[] = {4, 1, 4, 3, 1};
float output[output_length];
TF_LITE_MICRO_EXPECT_EQ(
kTfLiteError,
tflite::testing::TestSpaceToBatchNdFloat(
tflite::testing::basic_input_dims, tflite::testing::basic_input,
tflite::testing::basic_block_shape_dims,
tflite::testing::basic_block_shape, tflite::testing::basic_crops_dims,
tflite::testing::basic_crops, output_dims,
tflite::testing::basic_golden, output));
}
TF_LITE_MICRO_TESTS_END TF_LITE_MICRO_TESTS_END