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

Merge pull request #46903 from rsun-bdti:issue-45415-pr7

Browse Source 
PiperOrigin-RevId: 355767552
Change-Id: Ie79092e6be8c5069450e26023af173dfb7ca7a67
This commit is contained in:
TensorFlower Gardener 2021-02-04 20:10:27 -08:00
commit 511357bc37
1 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -12,6 +12,7 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/
#include <limits>
#include "tensorflow/lite/c/builtin_op_data.h"
#include "tensorflow/lite/c/common.h"
@ -49,7 +50,6 @@ void TestExp(const int* input_dims_data, const float* input_data,
TF_LITE_MICRO_EXPECT_EQ(kTfLiteOk, runner.InitAndPrepare());
TF_LITE_MICRO_EXPECT_EQ(kTfLiteOk, runner.Invoke());
for (int i = 0; i < output_dims_count; ++i) {
TF_LITE_MICRO_EXPECT_NEAR(expected_output_data[i], output_data[i], 1e-5f);
}
@ -63,14 +63,12 @@ TF_LITE_MICRO_TESTS_BEGIN
TF_LITE_MICRO_TEST(SingleDim) {
float output_data[7];
const int input_dims[] = {1, 1, 7};
const int input_dims[] = {2, 1, 7};
const float input_values[] = {0.0f, 1.0f, -1.0f, 100.0f,
-100.0f, 0.01f, -0.01f};
// (1.17549e-038 ~ 3.40282e+038), so the golden is set to float32's upper
// limit.
const float golden[] = {1.0f, 2.71823f, 0.36788f, 3.40282e+038f,
1.17549e-38f, 1.01005f, 0.99005f};
const float golden[] = {
1.0f, 2.71828f, 0.36788f, std::numeric_limits<float>::infinity(),
1.17549e-38f, 1.01005f, 0.99005f};
tflite::testing::TestExp(input_dims, input_values, golden, output_data);
}