diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-02-05 14:43:35 -0800 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-05 14:47:27 -0800 |
| commit | 1c762f70caf7004470cdfa599b4eb7a76e5bcc78 (patch) | |
| tree | afaca56c1371ad158d943a4b3e8a69b26840f8ac /tensorflow/core/grappler/utils_test.cc | |
| parent | d53202bc9498a9045747282530380f92b4d66a00 (diff) | |
Backward pass implementation for fusion optimizer.
PiperOrigin-RevId: 184589487
Diffstat (limited to 'tensorflow/core/grappler/utils_test.cc')
| -rw-r--r-- | tensorflow/core/grappler/utils_test.cc | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/tensorflow/core/grappler/utils_test.cc b/tensorflow/core/grappler/utils_test.cc index 77371c399e..ba4e6b1bae 100644 --- a/tensorflow/core/grappler/utils_test.cc +++ b/tensorflow/core/grappler/utils_test.cc @@ -177,9 +177,10 @@ TEST_F(UtilsTest, ExecuteWithTimeout) { } TEST_F(UtilsTest, NumOutputs) { - EXPECT_EQ(2, NumOutputs(CreateConcatOffsetNode())); - EXPECT_EQ(5, NumOutputs(CreateFusedBatchNormNode())); - EXPECT_EQ(1, NumOutputs(CreateDequeueNode())); + GraphDef graph; + EXPECT_EQ(2, NumOutputs(CreateConcatOffsetNode(), &graph)); + EXPECT_EQ(5, NumOutputs(CreateFusedBatchNormNode(), &graph)); + EXPECT_EQ(1, NumOutputs(CreateDequeueNode(), &graph)); } TEST_F(UtilsTest, AsControlDependency) { |