diff options
author | 2017-02-01 22:38:30 -0800 | |
---|---|---|
committer | 2017-02-01 22:46:33 -0800 | |
commit | 4957f8a8de587ff12d367ead7d91479005487a08 (patch) | |
tree | c17f14efe95a43bc0bd4551d79da1e74660e1528 /tensorflow/cc/gradients/array_grad_test.cc | |
parent | c1238db44e3344b661ae3e6aa17ff74898569b11 (diff) |
Update generated C++ API. This is a roll-forward past CLs that were
commented out plus some new overrides.
Change: 146330232
Diffstat (limited to 'tensorflow/cc/gradients/array_grad_test.cc')
-rw-r--r-- | tensorflow/cc/gradients/array_grad_test.cc | 20 |
1 files changed, 10 insertions, 10 deletions
diff --git a/tensorflow/cc/gradients/array_grad_test.cc b/tensorflow/cc/gradients/array_grad_test.cc index 0cdb3b7a1f..5798b5b509 100644 --- a/tensorflow/cc/gradients/array_grad_test.cc +++ b/tensorflow/cc/gradients/array_grad_test.cc @@ -53,41 +53,41 @@ class ArrayGradTest : public ::testing::Test { Scope scope_; }; -TEST_F(ArrayGradTest, PackGrad_Axis0) { +TEST_F(ArrayGradTest, StackGrad_Axis0) { TensorShape x_shape({1, 2, 3}); std::vector<Output> xs; xs.push_back(Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x_shape))); xs.push_back(Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x_shape))); - auto y = Pack(scope_, xs, Pack::Axis(0)); + auto y = Stack(scope_, xs, Stack::Axis(0)); TensorShape y_shape({2, 1, 2, 3}); RunTest(xs, {x_shape, x_shape}, {y}, {y_shape}); } -TEST_F(ArrayGradTest, PackGrad_Axis1) { +TEST_F(ArrayGradTest, StackGrad_Axis1) { TensorShape x_shape({1, 2, 3}); std::vector<Output> xs; xs.push_back(Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x_shape))); xs.push_back(Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x_shape))); - auto y = Pack(scope_, xs, Pack::Axis(1)); + auto y = Stack(scope_, xs, Stack::Axis(1)); TensorShape y_shape({1, 2, 2, 3}); RunTest(xs, {x_shape, x_shape}, {y}, {y_shape}); } -TEST_F(ArrayGradTest, UnpackGrad_Axis0) { +TEST_F(ArrayGradTest, UnstackGrad_Axis0) { TensorShape x_shape({4, 2, 3}); auto x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x_shape)); - // Unpacking the first dimension results in 4 outputs. + // Unstacking the first dimension results in 4 outputs. std::vector<TensorShape> y_shapes(4, TensorShape({2, 3})); - auto y = Unpack(scope_, x, 4, Unpack::Axis(0)); + auto y = Unstack(scope_, x, 4, Unstack::Axis(0)); RunTest({x}, {x_shape}, y.output, y_shapes); } -TEST_F(ArrayGradTest, UnpackGrad_Axis1) { +TEST_F(ArrayGradTest, UnstackGrad_Axis1) { TensorShape x_shape({4, 2, 3}); auto x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x_shape)); - // Unpacking the second dimension results in 2 outputs. + // Unstacking the second dimension results in 2 outputs. std::vector<TensorShape> y_shapes(2, TensorShape({4, 3})); - auto y = Unpack(scope_, x, 2, Unpack::Axis(1)); + auto y = Unstack(scope_, x, 2, Unstack::Axis(1)); RunTest({x}, {x_shape}, y.output, y_shapes); } |