aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/builtin_op_data.h
diff options
context:
space:
mode:
authorGravatar Nupur Garg <nupurgarg@google.com>2017-12-13 14:54:10 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-12-13 14:57:35 -0800
commit57600a8d7739f6fbea445c6efa1f29f12f769748 (patch)
tree428b084408ba5847662056980a8fdc2c020b6dd1 /tensorflow/contrib/lite/builtin_op_data.h
parent7fbcd3724379161e379f4693ba66666b6ac4b0aa (diff)
Add support for "Pad".
PiperOrigin-RevId: 178962340
Diffstat (limited to 'tensorflow/contrib/lite/builtin_op_data.h')
-rw-r--r--tensorflow/contrib/lite/builtin_op_data.h8
1 files changed, 8 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/builtin_op_data.h b/tensorflow/contrib/lite/builtin_op_data.h
index 93072bf90b..7249d124e9 100644
--- a/tensorflow/contrib/lite/builtin_op_data.h
+++ b/tensorflow/contrib/lite/builtin_op_data.h
@@ -133,6 +133,14 @@ typedef struct {
typedef struct {
// 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 before_padding[8];
+ int after_padding[8];
+ int num_dimensions;
+} TfLitePadParams;
+
+typedef struct {
+ // 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 shape[8];
int num_dimensions;
} TfLiteReshapeParams;