Further improve performance of ParseExample fixing regression on specific benchmarks.

- Ensure that each thread receives equal piece.
- At least 8 threads (given 8 or more examples) and at most 64
- Use InlinedVector
Change: 132488652

1 files changed, 9 insertions, 0 deletions

diff --git a/tensorflow/core/util/example_proto_fast_parsing_test.cc b/tensorflow/core/util/example_proto_fast_parsing_test.cc
index 6d3b548851..9da3f6ad2a 100644
--- a/tensorflow/core/util/example_proto_fast_parsing_test.cc
+++ b/tensorflow/core/util/example_proto_fast_parsing_test.cc
@@ -178,6 +178,15 @@ string MakeSerializedExample() {
   return serialized;
 }
 
+TEST(TestFastParseExample, Empty) {
+  Result result;
+  FastParseExampleConfig config;
+  config.sparse.push_back({"test", DT_STRING});
+  Status status = FastParseExample(config, gtl::ArraySlice<string>(),
+                                   gtl::ArraySlice<string>(), nullptr, &result);
+  EXPECT_TRUE(status.ok()) << status;
+}
+
 }  // namespace
 
 }  // namespace example