aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/cc/gradients/array_grad_test.cc
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2017-02-01 22:38:30 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-02-01 22:46:33 -0800
commit4957f8a8de587ff12d367ead7d91479005487a08 (patch)
treec17f14efe95a43bc0bd4551d79da1e74660e1528 /tensorflow/cc/gradients/array_grad_test.cc
parentc1238db44e3344b661ae3e6aa17ff74898569b11 (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.cc20
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);
}