aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/util/example_proto_fast_parsing.h
diff options
context:
space:
mode:
authorGravatar Patrik Sundberg <sundberg@google.com>2018-08-04 12:48:13 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-08-04 12:52:16 -0700
commitcb80a1ed9fafc9274f031adca4ec5b754ac93f2a (patch)
treecc5089e88f4d1b51fbb2f513ab09251e88308dec /tensorflow/core/util/example_proto_fast_parsing.h
parent3a41e5363530f058cb2b57cf0add09931ec788b2 (diff)
Add a batch sequence example parsing op, part 1.
PiperOrigin-RevId: 207406637
Diffstat (limited to 'tensorflow/core/util/example_proto_fast_parsing.h')
-rw-r--r--tensorflow/core/util/example_proto_fast_parsing.h11
1 files changed, 11 insertions, 0 deletions
diff --git a/tensorflow/core/util/example_proto_fast_parsing.h b/tensorflow/core/util/example_proto_fast_parsing.h
index 1b08f02267..024a4518ee 100644
--- a/tensorflow/core/util/example_proto_fast_parsing.h
+++ b/tensorflow/core/util/example_proto_fast_parsing.h
@@ -85,6 +85,17 @@ typedef FastParseExampleConfig FastParseSingleExampleConfig;
Status FastParseSingleExample(const FastParseSingleExampleConfig& config,
const string& serialized, Result* result);
+// Parses a batch of serialized SequenceExample protos and converts them into
+// result according to given config.
+// Given example names have to either be empty or the same size as serialized.
+// example_names are used only for error messages.
+Status FastParseSequenceExample(
+ const example::FastParseExampleConfig& context_config,
+ const example::FastParseExampleConfig& feature_list_config,
+ gtl::ArraySlice<string> serialized, gtl::ArraySlice<string> example_names,
+ thread::ThreadPool* thread_pool, example::Result* context_result,
+ example::Result* feature_list_result);
+
// This function parses serialized Example and populates given example.
// It uses the same specialized parser as FastParseExample which is efficient.
// But then constructs Example which is relatively slow.