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

Initialize tensors and re-enable tests.

Browse Source 
Avoid the ASAN failure due to unitialized tensors.

PiperOrigin-RevId: 209624108
This commit is contained in:
Jacques Pienaar 2018-08-21 11:00:35 -07:00 committed by TensorFlower Gardener
parent c72c36145b
commit 0d4212397d
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/compiler/tf2xla/functionalize_cond_test.cc
View File

@ -69,11 +69,12 @@ class FunctionalizeCondTest : public ::testing::Test {
namespace { namespace {
// TODO(jpienaar): Re-enable. Disabling for ASAN failure. TEST_F(FunctionalizeCondTest, ScopeIn) {
TEST_F(FunctionalizeCondTest, DISABLED_ScopeIn) {
Tensor pred_tensor(DT_BOOL, TensorShape()); Tensor pred_tensor(DT_BOOL, TensorShape());
pred_tensor.flat<bool>().setZero();
Node* pred = test::graph::Constant(graph_.get(), pred_tensor, "pred"); Node* pred = test::graph::Constant(graph_.get(), pred_tensor, "pred");
Tensor val_tensor(DT_INT32, TensorShape()); Tensor val_tensor(DT_INT32, TensorShape());
val_tensor.flat<int>().setZero();
Node* val = test::graph::Constant(graph_.get(), val_tensor, "val"); Node* val = test::graph::Constant(graph_.get(), val_tensor, "val");
Node* s = test::graph::Switch(graph_.get(), val, pred); Node* s = test::graph::Switch(graph_.get(), val, pred);
@ -112,11 +113,12 @@ TEST_F(FunctionalizeCondTest, DISABLED_ScopeIn) {
} }
} }
// TODO(jpienaar): Re-enable. Disabling for ASAN failure. TEST_F(FunctionalizeCondTest, JoinCondStates) {
TEST_F(FunctionalizeCondTest, DISABLED_JoinCondStates) {
Tensor pred_tensor(DT_BOOL, TensorShape()); Tensor pred_tensor(DT_BOOL, TensorShape());
pred_tensor.flat<bool>().setZero();
Node* pred = test::graph::Constant(graph_.get(), pred_tensor, "pred"); Node* pred = test::graph::Constant(graph_.get(), pred_tensor, "pred");
Tensor val_tensor(DT_INT32, TensorShape()); Tensor val_tensor(DT_INT32, TensorShape());
val_tensor.flat<int>().setZero();
Node* val = test::graph::Constant(graph_.get(), val_tensor, "val"); Node* val = test::graph::Constant(graph_.get(), val_tensor, "val");
Node* s = test::graph::Switch(graph_.get(), val, pred); Node* s = test::graph::Switch(graph_.get(), val, pred);