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

Make use of ::testing::TempDir to access tmp dir.

Browse Source 
PiperOrigin-RevId: 275115437
Change-Id: I10fcec58c9ebcc447f56306501447cab92cc29f3
This commit is contained in:
Brian Atkinson 2019-10-16 14:34:03 -07:00 committed by TensorFlower Gardener
parent 7874c7034c
commit 1c88fff548
2 changed files with 1 additions and 5 deletions
tensorflow/lite/testing/kernel_test
BUILDutil_test.cc

3
tensorflow/lite/testing/kernel_test/BUILD
View File

@ -38,9 +38,6 @@ tf_cc_test(
"//tensorflow/lite:testdata/add.bin", "//tensorflow/lite:testdata/add.bin",
"//tensorflow/lite:testdata/test_input.csv", "//tensorflow/lite:testdata/test_input.csv",
], ],
tags = [
"no_oss",
],
deps = [ deps = [
":util", ":util",
"//tensorflow/lite/testing:tflite_driver", "//tensorflow/lite/testing:tflite_driver",

3
tensorflow/lite/testing/kernel_test/util_test.cc
View File

@ -18,7 +18,6 @@ limitations under the License.
#include <memory> #include <memory>
#include <gmock/gmock.h> #include <gmock/gmock.h>
#include "testing/base/public/googletest.h"
#include <gtest/gtest.h> #include <gtest/gtest.h>
#include "tensorflow/lite/testing/tflite_driver.h" #include "tensorflow/lite/testing/tflite_driver.h"
@ -32,7 +31,7 @@ TEST(UtilTest, SimpleE2ETest) {
options.tflite_model = "tensorflow/lite/testdata/add.bin"; options.tflite_model = "tensorflow/lite/testdata/add.bin";
options.read_input_from_file = options.read_input_from_file =
"tensorflow/lite/testdata/test_input.csv"; "tensorflow/lite/testdata/test_input.csv";
options.dump_output_to_file = FLAGS_test_tmpdir + "/test_out.csv"; options.dump_output_to_file = ::testing::TempDir() + "/test_out.csv";
options.kernel_type = "REFERENCE"; options.kernel_type = "REFERENCE";
std::unique_ptr<TestRunner> runner(new TfLiteDriver( std::unique_ptr<TestRunner> runner(new TfLiteDriver(
TfLiteDriver::DelegateType::kNone, /*reference_kernel=*/true)); TfLiteDriver::DelegateType::kNone, /*reference_kernel=*/true));