diff options
author | 2018-03-08 11:56:29 -0800 | |
---|---|---|
committer | 2018-03-08 12:00:28 -0800 | |
commit | 6e3a43f4b7a1288c878b5daff274f1229256fbe8 (patch) | |
tree | 65dbe6000d5bf896015d64519a2d5783c1857e83 /tensorflow/contrib/lite/util.h | |
parent | 214ad0978641a946c25b334c4a33ecd1793b4d70 (diff) |
TFLite: Delegate Buffer Handle interface (take 2)
PiperOrigin-RevId: 188366045
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_ |