diff options
author | Jared Duke <jdduke@google.com> | 2018-07-26 12:35:26 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-26 12:38:50 -0700 |
commit | 20154feed54b667d5d9819aaa0d75626a6c02228 (patch) | |
tree | 29151b71ccd53b65a8c185ca4e9b2aaf44a74caf /tensorflow/contrib/lite/context.h | |
parent | 99deb5e0c561a926fc73c72d1488d20756f25861 (diff) |
More changes for TFLite portability
PiperOrigin-RevId: 206202981
Diffstat (limited to 'tensorflow/contrib/lite/context.h')
-rw-r--r-- | tensorflow/contrib/lite/context.h | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/context.h b/tensorflow/contrib/lite/context.h index cbfce12d7e..5bc20106d3 100644 --- a/tensorflow/contrib/lite/context.h +++ b/tensorflow/contrib/lite/context.h @@ -29,6 +29,9 @@ limitations under the License. #ifndef TENSORFLOW_CONTRIB_LITE_CONTEXT_H_ #define TENSORFLOW_CONTRIB_LITE_CONTEXT_H_ +#if defined(_MSC_VER) +#include <complex.h> +#endif #include <stdbool.h> #include <stdint.h> #include <stdlib.h> @@ -180,7 +183,11 @@ typedef union { uint8_t* uint8; bool* b; int16_t* i16; +#if defined(_MSC_VER) + _Fcomplex* c64; +#else _Complex float* c64; +#endif } TfLitePtrUnion; // Memory allocation strategies. kTfLiteMmapRo is for read-only memory-mapped |