aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/builtin_op_data.h
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-07-10 01:53:40 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-07-10 01:56:38 -0700
commit7f85c95f71b01f711c366942a7cd911b0743b72c (patch)
treead92abbdf8a4c5736e323a75c3c153dadf3ce5c6 /tensorflow/contrib/lite/builtin_op_data.h
parentc59bb780ebd1674ab34dd96d193c71698682ed4d (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.h4
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;