diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-06 14:56:18 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-06 15:07:58 -0700 |
| commit | 3b44d4bbfccce918ea9155e33c3da55c770b781f (patch) | |
| tree | c4ad72f378e1be32f6f3ff02ef72f4dd7bc83914 /tensorflow/contrib/lite/toco/tooling_util.h | |
| parent | e0a8285d9563122a75d94a54352f5c94f287e810 (diff) | |
Convert more kernel signatures to use runtime shapes.
PiperOrigin-RevId: 211874785
Diffstat (limited to 'tensorflow/contrib/lite/toco/tooling_util.h')
| -rw-r--r-- | tensorflow/contrib/lite/toco/tooling_util.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/toco/tooling_util.h b/tensorflow/contrib/lite/toco/tooling_util.h index bdeb203024..5f4b8cb66a 100644 --- a/tensorflow/contrib/lite/toco/tooling_util.h +++ b/tensorflow/contrib/lite/toco/tooling_util.h @@ -28,6 +28,7 @@ limitations under the License. #if TOCO_SUPPORT_PORTABLE_PROTOS #include "third_party/protobuf/include/google/protobuf/text_format.h" #endif // TOCO_SUPPORT_PORTABLE_PROTOS +#include "tensorflow/contrib/lite/kernels/internal/types.h" #include "tensorflow/contrib/lite/toco/model.h" #include "tensorflow/contrib/lite/toco/model_flags.pb.h" #include "tensorflow/contrib/lite/toco/runtime/types.h" @@ -139,6 +140,10 @@ bool ShapesAgreeUpToBroadcasting(const Shape& shape0, const Shape& shape1); // - For the remaining indices [0..i0), d0[i0] == 1. bool ShapesAgreeUpToExtending(const Shape& shape0, const Shape& shape1); +inline ::tflite::RuntimeShape ToRuntimeShape(const Shape& shape) { + return ::tflite::RuntimeShape(shape.dimensions_count(), shape.dims().data()); +} + bool IsArrayFullyConnectedWeights(const Model& model, const string& name); // If there is a wildcard dimension (-1), this may return a negative value. |