tflite_op_resolver.h 793 B

12345678910111213141516171819202122
  1. // Copyright 2021 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #ifndef COMPONENTS_OPTIMIZATION_GUIDE_CORE_TFLITE_OP_RESOLVER_H_
  5. #define COMPONENTS_OPTIMIZATION_GUIDE_CORE_TFLITE_OP_RESOLVER_H_
  6. #include "third_party/tflite/src/tensorflow/lite/model.h"
  7. #include "third_party/tflite/src/tensorflow/lite/mutable_op_resolver.h"
  8. namespace optimization_guide {
  9. // This class maintains all the currently supported TFLite
  10. // operations for the Chromium build of TFLite and registers them for use.
  11. class TFLiteOpResolver : public tflite::MutableOpResolver {
  12. public:
  13. TFLiteOpResolver();
  14. };
  15. } // namespace optimization_guide
  16. #endif // COMPONENTS_OPTIMIZATION_GUIDE_CORE_TFLITE_OP_RESOLVER_H_