aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/builtin_op_data.h
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-02-15 14:49:01 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-02-15 14:52:56 -0800
commit41a8560e0c2fa3b8fa73622a15da53f5e1b8b6c8 (patch)
tree4b97df5ca9ecff906c675ded62a29f7cefd52339 /tensorflow/contrib/lite/builtin_op_data.h
parent66f4f4cf31b86b7dd20f10ce6d968348b502f2ee (diff)
Implement Split
PiperOrigin-RevId: 185904437
Diffstat (limited to 'tensorflow/contrib/lite/builtin_op_data.h')
-rw-r--r--tensorflow/contrib/lite/builtin_op_data.h4
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/builtin_op_data.h b/tensorflow/contrib/lite/builtin_op_data.h
index 5dbeadd165..5fc8954743 100644
--- a/tensorflow/contrib/lite/builtin_op_data.h
+++ b/tensorflow/contrib/lite/builtin_op_data.h
@@ -196,6 +196,10 @@ typedef struct {
} TfLiteMeanParams;
typedef struct {
+ int num_splits;
+} TfLiteSplitParams;
+
+typedef struct {
// TODO(ahentz): We can't have dynamic data in this struct, at least not yet.
// For now we will fix the maximum possible number of dimensions.
int squeeze_dims[8];