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

Fix MSAN failure in delegate_test

Browse Source 
Added inital value of input0 in StaticOutputRFFT()

PiperOrigin-RevId: 334107533
Change-Id: I6e691aef799e942bccf2de667aafe97feb6daf63
This commit is contained in:
Terry Heo 2020-09-28 01:41:32 -07:00 committed by TensorFlower Gardener
parent 1c3145aeba
commit 516c0875a7
2 changed files with 2 additions and 1 deletions
tensorflow/lite/delegates/flex
BUILDdelegate_test.cc

1
tensorflow/lite/delegates/flex/BUILD
View File

@ -129,7 +129,6 @@ tf_cc_test(
srcs = ["delegate_test.cc"], srcs = ["delegate_test.cc"],
tags = [ tags = [
"no_gpu", # GPU + flex is not officially supported. "no_gpu", # GPU + flex is not officially supported.
"nomsan", # b/169431678
], ],
deps = [ deps = [
":delegate", ":delegate",

2
tensorflow/lite/delegates/flex/delegate_test.cc
View File

@ -15,6 +15,7 @@ limitations under the License.
#include "tensorflow/lite/delegates/flex/delegate.h" #include "tensorflow/lite/delegates/flex/delegate.h"
#include <cstdint> #include <cstdint>
#include <vector>
#include <gmock/gmock.h> #include <gmock/gmock.h>
#include <gtest/gtest.h> #include <gtest/gtest.h>
@ -361,6 +362,7 @@ TEST_F(DelegateTest, StaticOutputRFFT) {
// Define inputs. // Define inputs.
SetShape(0, {3, 512}); SetShape(0, {3, 512});
SetValues(0, std::vector<float>(3 * 512, 1.0f));
ASSERT_TRUE(Invoke()); ASSERT_TRUE(Invoke());