all_ops_resolver.h 1.4 KB

1234567891011121314151617181920212223242526272829303132333435
  1. /* Copyright 2018 The TensorFlow Authors. All Rights Reserved.
  2. Licensed under the Apache License, Version 2.0 (the "License");
  3. you may not use this file except in compliance with the License.
  4. You may obtain a copy of the License at
  5. http://www.apache.org/licenses/LICENSE-2.0
  6. Unless required by applicable law or agreed to in writing, software
  7. distributed under the License is distributed on an "AS IS" BASIS,
  8. WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  9. See the License for the specific language governing permissions and
  10. limitations under the License.
  11. ==============================================================================*/
  12. #ifndef TENSORFLOW_LITE_MICRO_ALL_OPS_RESOLVER_H_
  13. #define TENSORFLOW_LITE_MICRO_ALL_OPS_RESOLVER_H_
  14. #include "tensorflow/lite/micro/compatibility.h"
  15. #include "tensorflow/lite/micro/micro_mutable_op_resolver.h"
  16. namespace tflite {
  17. // The magic number in the template parameter is the maximum number of ops that
  18. // can be added to AllOpsResolver. It can be increased if needed. And most
  19. // applications that care about the memory footprint will want to directly use
  20. // MicroMutableOpResolver and have an application specific template parameter.
  21. // The examples directory has sample code for this.
  22. class AllOpsResolver : public MicroMutableOpResolver<128> {
  23. public:
  24. AllOpsResolver();
  25. private:
  26. TF_LITE_REMOVE_VIRTUAL_DELETE
  27. };
  28. } // namespace tflite
  29. #endif // TENSORFLOW_LITE_MICRO_ALL_OPS_RESOLVER_H_