diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-04-04 07:45:31 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-04 07:50:20 -0700 |
| commit | 3fbec92fa4997fc834807b57b229fa9ada179f6c (patch) | |
| tree | 34a7c15af0428e5ecf2ee2deb16e7c26b3d4320b /tensorflow/contrib/lite/builtin_op_data.h | |
| parent | 71b19430e8484b136e0b872f6a543aff8a242587 (diff) | |
Turn Cast into a proper builtin operator.
PiperOrigin-RevId: 191590230
Diffstat (limited to 'tensorflow/contrib/lite/builtin_op_data.h')
| -rw-r--r-- | tensorflow/contrib/lite/builtin_op_data.h | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/builtin_op_data.h b/tensorflow/contrib/lite/builtin_op_data.h index 5fc8954743..2b6c24768c 100644 --- a/tensorflow/contrib/lite/builtin_op_data.h +++ b/tensorflow/contrib/lite/builtin_op_data.h @@ -17,6 +17,8 @@ limitations under the License. #include <stdint.h> +#include "tensorflow/contrib/lite/context.h" + #ifdef __cplusplus extern "C" { #endif // __cplusplus @@ -174,6 +176,11 @@ typedef struct { int block_size; } TfLiteSpaceToDepthParams; +typedef struct { + TfLiteType in_data_type; + TfLiteType out_data_type; +} TfLiteCastParams; + typedef enum { kTfLiteCombinerTypeSum = 0, kTfLiteCombinerTypeMean = 1, |