micro_optional_debug_tools.h

00001 /* Copyright 2017 The TensorFlow Authors. All Rights Reserved.
00002 
00003 Licensed under the Apache License, Version 2.0 (the "License");
00004 you may not use this file except in compliance with the License.
00005 You may obtain a copy of the License at
00006 
00007     http://www.apache.org/licenses/LICENSE-2.0
00008 
00009 Unless required by applicable law or agreed to in writing, software
00010 distributed under the License is distributed on an "AS IS" BASIS,
00011 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
00012 See the License for the specific language governing permissions and
00013 limitations under the License.
00014 ==============================================================================*/
00015 // Optional debugging functionality. For small sized binaries, these are not
00016 // needed.
00017 #ifndef TENSORFLOW_LITE_MICRO_OPTIONAL_DEBUG_TOOLS_H_
00018 #define TENSORFLOW_LITE_MICRO_OPTIONAL_DEBUG_TOOLS_H_
00019 
00020 #include "tensorflow/lite/experimental/micro/micro_interpreter.h"
00021 
00022 namespace tflite {
00023 // Prints a dump of what tensors and what nodes are in the interpreter.
00024 class MicroInterpreter;
00025 void PrintInterpreterState(MicroInterpreter* interpreter);
00026 
00027 #ifdef __cplusplus
00028 extern "C" {
00029 #endif  // __cplusplus
00030 struct pairTfLiteNodeAndRegistration {
00031   TfLiteNode node;
00032   const TfLiteRegistration* registration;
00033 };
00034 #ifdef __cplusplus
00035 }  // extern "C"
00036 #endif  // __cplusplus
00037 
00038 }  // namespace tflite
00039 
00040 #endif  // TENSORFLOW_LITE_MICRO_OPTIONAL_DEBUG_TOOLS_H_