diff options
author | 2017-12-14 14:48:02 -0800 | |
---|---|---|
committer | 2017-12-14 14:55:51 -0800 | |
commit | ea703f4e0e72d1e016f8157e206dcc9e80602862 (patch) | |
tree | 092e759fbe5118413efc3cc37f02a008ec9f2711 /tensorflow/contrib/lite/builtin_op_data.h | |
parent | ab54c4885366e87bd2805036c9c8ab3ffccf823e (diff) |
Implementation of Gather in TfLite
PiperOrigin-RevId: 179101363
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 7249d124e9..548864a1e9 100644 --- a/tensorflow/contrib/lite/builtin_op_data.h +++ b/tensorflow/contrib/lite/builtin_op_data.h @@ -165,6 +165,10 @@ typedef struct { TfLiteCombinerType combiner; } TfLiteEmbeddingLookupSparseParams; +typedef struct { + int axis; +} TfLiteGatherParams; + #ifdef __cplusplus } // extern "C" #endif // __cplusplus |