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_ops.h | |
parent | 1c8f3c81698b67b8fffce86c97df27d392b84cb8 (diff) |
Implementation of ArgMax
PiperOrigin-RevId: 192236845
Diffstat (limited to 'tensorflow/contrib/lite/builtin_ops.h')
-rw-r--r-- | tensorflow/contrib/lite/builtin_ops.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/builtin_ops.h b/tensorflow/contrib/lite/builtin_ops.h index 17b791e4e2..e11c7fb2e4 100644 --- a/tensorflow/contrib/lite/builtin_ops.h +++ b/tensorflow/contrib/lite/builtin_ops.h @@ -80,6 +80,7 @@ typedef enum { kTfLiteBuiltinCast = 53, kTfLiteBuiltinPrelu = 54, kTfLiteBuiltinMaximum = 55, + kTfLiteBuiltinArgMax = 56, } TfLiteBuiltinOperator; #ifdef __cplusplus |