aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/builtin_op_data.h
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-08-23 21:50:34 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-08-23 21:57:27 -0700
commit4bc1d3e484c6eb3ea2ba4e6400722be32220c808 (patch)
treeb6f760b3003355257f57b2441ac273e7349e3b03 /tensorflow/contrib/lite/builtin_op_data.h
parent0c657f3b9f6ef6ee63b3eb54fe928f482c58dc80 (diff)
Implementation of unpack op.
PiperOrigin-RevId: 210051131
Diffstat (limited to 'tensorflow/contrib/lite/builtin_op_data.h')
-rw-r--r--tensorflow/contrib/lite/builtin_op_data.h5
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/builtin_op_data.h b/tensorflow/contrib/lite/builtin_op_data.h
index 70178b2faa..e81f9e4f51 100644
--- a/tensorflow/contrib/lite/builtin_op_data.h
+++ b/tensorflow/contrib/lite/builtin_op_data.h
@@ -286,6 +286,11 @@ typedef struct {
int axis;
} TfLiteOneHotParams;
+typedef struct {
+ int num;
+ int axis;
+} TfLiteUnpackParams;
+
#ifdef __cplusplus
} // extern "C"
#endif // __cplusplus