diff options
Diffstat (limited to 'tensorflow/contrib/lite/builtin_op_data.h')
-rw-r--r-- | tensorflow/contrib/lite/builtin_op_data.h | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/builtin_op_data.h b/tensorflow/contrib/lite/builtin_op_data.h index cda889bf50..fd16aa1063 100644 --- a/tensorflow/contrib/lite/builtin_op_data.h +++ b/tensorflow/contrib/lite/builtin_op_data.h @@ -250,6 +250,10 @@ typedef struct { } TfLiteArgMaxParams; typedef struct { + TfLiteType output_type; +} TfLiteArgMinParams; + +typedef struct { TfLitePadding padding; int stride_width; int stride_height; @@ -263,6 +267,21 @@ typedef struct { TfLiteType out_type; } TfLiteShapeParams; +typedef struct { + // Parameters supported by version 1: + float min; + float max; + int num_bits; + + // Parameters supported by version 2: + bool narrow_range; +} TfLiteFakeQuantParams; + +typedef struct { + int values_count; + int axis; +} TfLitePackParams; + #ifdef __cplusplus } // extern "C" #endif // __cplusplus |