diff options
author | 2018-01-11 14:01:15 -0800 | |
---|---|---|
committer | 2018-01-11 14:05:06 -0800 | |
commit | a663e5fda16c0c739eb11279422549d499bad597 (patch) | |
tree | 432980ff6599fa78f30ef4bdbae567972fffd737 /tensorflow/contrib/lite/builtin_op_data.h | |
parent | 57fe4920932ccdfea3ce2235823cfb4327d119e7 (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.h | 8 |
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 |