diff options
author | 2018-04-09 22:01:59 -0700 | |
---|---|---|
committer | 2018-04-09 22:04:16 -0700 | |
commit | 3f3a6e6685449130389e0b8d76f6ba0fc457bcfe (patch) | |
tree | 6634bcc3cc6ad37e999de22a828c5f120b3d7082 /tensorflow/contrib/lite/builtin_op_data.h | |
parent | 1c8f3c81698b67b8fffce86c97df27d392b84cb8 (diff) |
Implementation of ArgMax
PiperOrigin-RevId: 192236845
Diffstat (limited to 'tensorflow/contrib/lite/builtin_op_data.h')
-rw-r--r-- | tensorflow/contrib/lite/builtin_op_data.h | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/builtin_op_data.h b/tensorflow/contrib/lite/builtin_op_data.h index 2b6c24768c..f5fb2f15e3 100644 --- a/tensorflow/contrib/lite/builtin_op_data.h +++ b/tensorflow/contrib/lite/builtin_op_data.h @@ -221,6 +221,10 @@ typedef struct { int shrink_axis_mask; } TfLiteStridedSliceParams; +typedef struct { + TfLiteType output_type; +} TfLiteArgMaxParams; + #ifdef __cplusplus } // extern "C" #endif // __cplusplus |