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

Merge pull request #46962 from advaitjain:fix-exp-test-xtensa

Browse Source 
PiperOrigin-RevId: 355944394
Change-Id: I52b83d393ea4e3b55cf73a8d0c73bd855a73090a
This commit is contained in:
TensorFlower Gardener 2021-02-05 15:48:16 -08:00
commit 7dbb6cf44a
3 changed files with 26 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tensorflow/lite/micro/kernels/BUILD
View File

@ -235,6 +235,19 @@ tflite_micro_cc_test(
],
)
tflite_micro_cc_test(
name = "exp_test",
srcs = ["exp_test.cc"],
deps = [
":kernel_runner",
"//tensorflow/lite/c:common",
"//tensorflow/lite/micro:debug_log",
"//tensorflow/lite/micro:op_resolvers",
"//tensorflow/lite/micro:test_helpers",
"//tensorflow/lite/micro/testing:micro_test",
],
)
tflite_micro_cc_test(
name = "pooling_test",
srcs = [

16
tensorflow/lite/micro/kernels/exp_test.cc
View File

@ -54,7 +54,6 @@ void TestExp(const int* input_dims_data, const float* input_data,
TF_LITE_MICRO_EXPECT_NEAR(expected_output_data[i], output_data[i], 1e-5f);
}
}
} // namespace
} // namespace testing
} // namespace tflite
@ -62,13 +61,16 @@ void TestExp(const int* input_dims_data, const float* input_data,
TF_LITE_MICRO_TESTS_BEGIN
TF_LITE_MICRO_TEST(SingleDim) {
float output_data[7];
const int input_dims[] = {2, 1, 7};
const float input_values[] = {0.0f, 1.0f, -1.0f, 100.0f,
constexpr int kInputSize = 7;
float output_data[kInputSize];
const int input_dims[] = {2, 1, kInputSize};
const float input_values[kInputSize] = {0.0f, 1.0f, -1.0f, 100.0f,
-100.0f, 0.01f, -0.01f};
const float golden[] = {
1.0f, 2.71828f, 0.36788f, std::numeric_limits<float>::infinity(),
1.17549e-38f, 1.01005f, 0.99005f};
float golden[kInputSize];
for (int i = 0; i < kInputSize; ++i) {
golden[i] = std::exp(input_values[i]);
}
tflite::testing::TestExp(input_dims, input_values, golden, output_data);
}

4
tensorflow/lite/micro/testing/micro_test.h
View File

@ -142,12 +142,14 @@ extern bool did_test_fail;
} \
} while (false)
// The check vx != vy is needed to properly handle the case where both
// x and y evaluate to infinity. See #46960 for more details.
#define TF_LITE_MICRO_EXPECT_NEAR(x, y, epsilon) \
do { \
auto vx = (x); \
auto vy = (y); \
auto delta = ((vx) > (vy)) ? ((vx) - (vy)) : ((vy) - (vx)); \
if (delta > epsilon) { \
if (vx != vy && delta > epsilon) { \
MicroPrintf(#x " (%f) near " #y " (%f) failed at %s:%d", \
static_cast<double>(vx), static_cast<double>(vy), __FILE__, \
__LINE__); \