diff options
| author |  Geoffrey Irving <geoffreyi@google.com> | 2017-05-16 17:01:31 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-05-16 17:05:15 -0700 |
| commit | 43db5c623f748b6f9704e9e9be5a5a11fa2a4c1a (patch) | |
| tree | 985844ec8f6653f36e38592f9700dcaba66d94f2 /tensorflow/cc/ops | |
| parent | 7ab0c2eff12ea79648f6717dae8558d6669e5c27 (diff) | |
Automated g4 rollback of changelist 156244933
PiperOrigin-RevId: 156251356
Diffstat (limited to 'tensorflow/cc/ops')
| -rw-r--r-- | tensorflow/cc/ops/const_op_test.cc | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/cc/ops/const_op_test.cc b/tensorflow/cc/ops/const_op_test.cc index 3184edeb33..5a4770f879 100644 --- a/tensorflow/cc/ops/const_op_test.cc +++ b/tensorflow/cc/ops/const_op_test.cc @@ -28,9 +28,9 @@ void ExpectNodeEqual(const Node* n, gtl::ArraySlice<T> values, TensorShape shape) { EXPECT_TRUE(n->IsConstant()); Tensor tensor; - TF_EXPECT_OK(GetNodeAttr(n->attrs(), "value", &tensor)); + TF_EXPECT_OK(GetNodeAttr(n->def(), "value", &tensor)); DataType dtype; - TF_EXPECT_OK(GetNodeAttr(n->attrs(), "dtype", &dtype)); + TF_EXPECT_OK(GetNodeAttr(n->def(), "dtype", &dtype)); EXPECT_EQ(tensor.dtype(), dtype); test::ExpectTensorEqual<T>(tensor, test::AsTensor(values, shape)); } @@ -39,9 +39,9 @@ void ExpectTypeAndShape(const Node* n, DataType expected_dtype, TensorShape expected_shape) { EXPECT_TRUE(n->IsConstant()); Tensor tensor; - TF_EXPECT_OK(GetNodeAttr(n->attrs(), "value", &tensor)); + TF_EXPECT_OK(GetNodeAttr(n->def(), "value", &tensor)); DataType dtype; - TF_EXPECT_OK(GetNodeAttr(n->attrs(), "dtype", &dtype)); + TF_EXPECT_OK(GetNodeAttr(n->def(), "dtype", &dtype)); EXPECT_EQ(dtype, expected_dtype); EXPECT_EQ(expected_shape, TensorShape(tensor.shape())); } |