diff options
author | Yu-Cheng Ling <ycling@google.com> | 2018-03-07 17:42:08 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-07 17:46:14 -0800 |
commit | 5594bc3c43f6829b7ea77f96852c98fb41e4deb2 (patch) | |
tree | a7d2f5771c1758ec5a2fbf69ebe79c8bab6d9693 /tensorflow/contrib/lite/util.h | |
parent | 9cdfd3878935fb6c3c2a5da7f65ee0db6c751170 (diff) |
TFLite: Delegate Buffer Handle interface
PiperOrigin-RevId: 188263046
Diffstat (limited to 'tensorflow/contrib/lite/util.h')
-rw-r--r-- | tensorflow/contrib/lite/util.h | 34 |
1 files changed, 34 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/util.h b/tensorflow/contrib/lite/util.h new file mode 100644 index 0000000000..50e4fb839e --- /dev/null +++ b/tensorflow/contrib/lite/util.h @@ -0,0 +1,34 @@ +/* Copyright 2017 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +==============================================================================*/ + +// This file provides general C++ utility functions in TFLite. +// For example: Converting between `TfLiteIntArray`, `std::vector` and +// Flatbuffer vectors. These functions can't live in `context.h` since it's pure +// C. + +#ifndef TENSORFLOW_CONTRIB_LITE_UTIL_H_ +#define TENSORFLOW_CONTRIB_LITE_UTIL_H_ + +#include <vector> +#include "tensorflow/contrib/lite/context.h" + +namespace tflite { + +// Converts a `std::vector` to a `TfLiteIntArray`. +TfLiteIntArray* ConvertVectorToTfLiteIntArray(const std::vector<int>& input); + +} // namespace tflite + +#endif // TENSORFLOW_CONTRIB_LITE_UTIL_H_ |