diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-07-10 13:26:21 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-10 13:32:56 -0700 |
| commit | 89dab281fdf9156af098937cadb058d29fa05227 (patch) | |
| tree | 75fa580bd2fffdbd7c2c50cf7d87892465d26d33 /tensorflow/contrib/lite/kernels/select.cc | |
| parent | dd1ce0fd8f8c97728475428f562825480e9d4194 (diff) | |
Adding int16 as a supported Select type.
PiperOrigin-RevId: 204001217
Diffstat (limited to 'tensorflow/contrib/lite/kernels/select.cc')
| -rw-r--r-- | tensorflow/contrib/lite/kernels/select.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/select.cc b/tensorflow/contrib/lite/kernels/select.cc index 9b6cee3cb5..3cdb5db209 100644 --- a/tensorflow/contrib/lite/kernels/select.cc +++ b/tensorflow/contrib/lite/kernels/select.cc @@ -89,6 +89,9 @@ TfLiteStatus SelectEval(TfLiteContext* context, TfLiteNode* node) { case kTfLiteUInt8: \ TF_LITE_SELECT(uint8_t, op); \ break; \ + case kTfLiteInt16: \ + TF_LITE_SELECT(int16_t, op); \ + break; \ case kTfLiteInt32: \ TF_LITE_SELECT(int32_t, op); \ break; \ |