diff options
author | Pete Warden <petewarden@google.com> | 2018-09-24 15:54:32 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-24 16:02:13 -0700 |
commit | 1ff157d82dac29f5a3a3197b2664208f6ed6ba06 (patch) | |
tree | c751f5a665a27c660809c4884eb07f69b4983244 /tensorflow/contrib/lite/kernels/kernel_util.cc | |
parent | 9c58005ec86297a1d0a17dc4f7ad7cbae9c47e4b (diff) |
Portability preparation for more cross-platform prototyping.
PiperOrigin-RevId: 214346240
Diffstat (limited to 'tensorflow/contrib/lite/kernels/kernel_util.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/kernel_util.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/kernel_util.cc b/tensorflow/contrib/lite/kernels/kernel_util.cc index 08f942c933..503ef28459 100644 --- a/tensorflow/contrib/lite/kernels/kernel_util.cc +++ b/tensorflow/contrib/lite/kernels/kernel_util.cc @@ -107,6 +107,9 @@ bool HaveSameShapes(const TfLiteTensor* input1, const TfLiteTensor* input2) { return TfLiteIntArrayEqual(input1->dims, input2->dims); } +// TODO(petewarden): Having macros around this is ugly, look at other strategies +// before replicating this approach elsewhere. +#ifndef TF_LITE_STATIC_MEMORY TfLiteStatus CalculateShapeForBroadcast(TfLiteContext* context, const TfLiteTensor* input1, const TfLiteTensor* input2, @@ -125,5 +128,6 @@ TfLiteStatus CalculateShapeForBroadcast(TfLiteContext* context, *output_shape = shape.release(); return kTfLiteOk; } +#endif // TF_LITE_STATIC_MEMORY } // namespace tflite |