diff options
author | 2017-12-14 21:26:55 -0800 | |
---|---|---|
committer | 2017-12-14 21:31:06 -0800 | |
commit | dbcb1ffcca6a3c52e3c109a1739018350bc41925 (patch) | |
tree | aa4f440d1f8f4382cea7a1b4e4cfc623da2a76f0 /tensorflow/contrib/lite/builtin_op_data.h | |
parent | f806269602219d5095265d036f294cc9a6260971 (diff) |
Support BatchToSpaceND in TFLite
The internal implementation only support 4D tensors for now.
The dimension has to be 1 batch + 2 spatial + 1 other.
The most common format within this restriction is NHWC.
Cropping is not supported by the internal implementation.
PiperOrigin-RevId: 179143332
Diffstat (limited to 'tensorflow/contrib/lite/builtin_op_data.h')
-rw-r--r-- | tensorflow/contrib/lite/builtin_op_data.h | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/builtin_op_data.h b/tensorflow/contrib/lite/builtin_op_data.h index 548864a1e9..5c6f3016b1 100644 --- a/tensorflow/contrib/lite/builtin_op_data.h +++ b/tensorflow/contrib/lite/builtin_op_data.h @@ -105,6 +105,17 @@ typedef struct { } TfLiteAddParams; typedef struct { + // Number of spatial dimensions. + // For now only NHWC is supported, and the value should always be 2. + int num_spatial_dimensions; + // TODO(ahentz): We can't have dynamic data in this struct, at least not yet. + // For now we will fix the maximum possible number of dimensions. + int block_shape[2]; + int before_crops[2]; + int after_crops[2]; +} TfLiteBatchToSpaceNDParams; + +typedef struct { TfLiteFusedActivation activation; } TfLiteMulParams; |