Add C++ shape inference functions for ParseExample and

ParseSingleSequenceExample.

The attribute parsing code was moved over verbatim to example_proto_helper.*.

Change TF_CHECK_OK to TF_ASSERT_OK in all shape inference tests under core.
Change: 130041336

1 files changed, 9 insertions, 8 deletions

diff --git a/tensorflow/core/ops/image_ops_test.cc b/tensorflow/core/ops/image_ops_test.cc
index fc9640ffb4..1b5db82ab8 100644
--- a/tensorflow/core/ops/image_ops_test.cc
+++ b/tensorflow/core/ops/image_ops_test.cc
@@ -18,6 +18,7 @@ limitations under the License.
 #include "tensorflow/core/framework/op.h"
 #include "tensorflow/core/framework/shape_inference_testutil.h"
 #include "tensorflow/core/framework/tensor_testutil.h"
+#include "tensorflow/core/lib/core/status_test_util.h"
 #include "tensorflow/core/platform/test.h"
 
 namespace tensorflow {
@@ -61,17 +62,17 @@ TEST(ImageOpsTest, DecodeImage_ShapeFn) {
     INFER_OK(op, "[]", "[?,?,?]");
 
     // Set the channel and so that part of output shape is known.
-    TF_CHECK_OK(NodeDefBuilder("test", op_name)
-                    .Input({"a", 0, DT_STRING})
-                    .Attr("channels", 4)
-                    .Finalize(&op.node_def));
+    TF_ASSERT_OK(NodeDefBuilder("test", op_name)
+                     .Input({"a", 0, DT_STRING})
+                     .Attr("channels", 4)
+                     .Finalize(&op.node_def));
     INFER_OK(op, "[]", "[?,?,4]");
 
     // Negative channel value is rejected.
-    TF_CHECK_OK(NodeDefBuilder("test", op_name)
-                    .Input({"a", 0, DT_STRING})
-                    .Attr("channels", -1)
-                    .Finalize(&op.node_def));
+    TF_ASSERT_OK(NodeDefBuilder("test", op_name)
+                     .Input({"a", 0, DT_STRING})
+                     .Attr("channels", -1)
+                     .Finalize(&op.node_def));
     INFER_ERROR("channels must be non-negative, got -1", op, "[]");
   }
 }