diff options
author | 2018-01-18 13:31:04 -0800 | |
---|---|---|
committer | 2018-01-18 13:34:51 -0800 | |
commit | 45c47cabe7150386420b182a8026699ff704b8f4 (patch) | |
tree | 675246e399521be14e0f369c3507ad3d3bd5f673 /tensorflow/contrib/lite/builtin_op_data.h | |
parent | 053470bc1a06b5f1cc65605bf21d48b3e92d6857 (diff) |
Supports Squeeze in Tf Lite.
PiperOrigin-RevId: 182429180
Diffstat (limited to 'tensorflow/contrib/lite/builtin_op_data.h')
-rw-r--r-- | tensorflow/contrib/lite/builtin_op_data.h | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/builtin_op_data.h b/tensorflow/contrib/lite/builtin_op_data.h index b9d88128c2..0c333f9e8c 100644 --- a/tensorflow/contrib/lite/builtin_op_data.h +++ b/tensorflow/contrib/lite/builtin_op_data.h @@ -214,6 +214,13 @@ typedef struct { bool keep_dims; } TfLiteMeanParams; +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]; + int num_squeeze_dims; +} TfLiteSqueezeParams; + #ifdef __cplusplus } // extern "C" #endif // __cplusplus |