aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/builtin_op_data.h
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-01-11 14:01:15 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-01-11 14:05:06 -0800
commita663e5fda16c0c739eb11279422549d499bad597 (patch)
tree432980ff6599fa78f30ef4bdbae567972fffd737 /tensorflow/contrib/lite/builtin_op_data.h
parent57fe4920932ccdfea3ce2235823cfb4327d119e7 (diff)
Adds Mean op to Tensorflow Lite.
PiperOrigin-RevId: 181658399
Diffstat (limited to 'tensorflow/contrib/lite/builtin_op_data.h')
-rw-r--r--tensorflow/contrib/lite/builtin_op_data.h8
1 files changed, 8 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/builtin_op_data.h b/tensorflow/contrib/lite/builtin_op_data.h
index 347e46b83c..062fea2aa3 100644
--- a/tensorflow/contrib/lite/builtin_op_data.h
+++ b/tensorflow/contrib/lite/builtin_op_data.h
@@ -198,6 +198,14 @@ typedef struct {
int num_dimensions;
} TfLiteTransposeParams;
+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 axis[8];
+ int num_axis_dimensions;
+ bool keep_dims;
+} TfLiteMeanParams;
+
#ifdef __cplusplus
} // extern "C"
#endif // __cplusplus