diff options
Diffstat (limited to 'tensorflow/contrib/lite/kernels/register.h')
-rw-r--r-- | tensorflow/contrib/lite/kernels/register.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/kernels/register.h b/tensorflow/contrib/lite/kernels/register.h index 0296152d68..61856ab9de 100644 --- a/tensorflow/contrib/lite/kernels/register.h +++ b/tensorflow/contrib/lite/kernels/register.h @@ -16,8 +16,9 @@ limitations under the License. #define TENSORFLOW_CONTRIB_LITE_KERNELS_REGISTER_H_ #include <unordered_map> -#include "tensorflow/contrib/lite/context.h" +#include "tensorflow/contrib/lite/c/c_api_internal.h" #include "tensorflow/contrib/lite/model.h" +#include "tensorflow/contrib/lite/mutable_op_resolver.h" namespace tflite { namespace ops { |