diff options
author | 2018-07-10 01:53:40 -0700 | |
---|---|---|
committer | 2018-07-10 01:56:38 -0700 | |
commit | 7f85c95f71b01f711c366942a7cd911b0743b72c (patch) | |
tree | ad92abbdf8a4c5736e323a75c3c153dadf3ce5c6 /tensorflow/contrib/lite/builtin_op_data.h | |
parent | c59bb780ebd1674ab34dd96d193c71698682ed4d (diff) |
Implementation of arg_min.
PiperOrigin-RevId: 203908601
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 cda889bf50..31c6cdfedb 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; |