diff --git a/tflite/toco/tflite/export.cc b/tflite/toco/tflite/export.cc
index c7a56146..259f6666 100644
--- a/tflite/toco/tflite/export.cc
+++ b/tflite/toco/tflite/export.cc
@@ -14,7 +14,12 @@ limitations under the License.
 ==============================================================================*/
 #include "tflite/toco/tflite/export.h"
 
+#include <cstdint>
+#include <map>
+#include <memory>
+#include <set>
 #include <string>
+#include <vector>
 
 #include "flatbuffers/flexbuffers.h"
 #include "absl/log/log.h"
diff --git a/tflite/toco/tflite/export.h b/tflite/toco/tflite/export.h
index e9af5016..ed3bce04 100644
--- a/tflite/toco/tflite/export.h
+++ b/tflite/toco/tflite/export.h
@@ -15,7 +15,12 @@ limitations under the License.
 #ifndef TENSORFLOW_LITE_TOCO_TFLITE_EXPORT_H_
 #define TENSORFLOW_LITE_TOCO_TFLITE_EXPORT_H_
 
+#include <cstddef>
+#include <functional>
+#include <map>
+#include <memory>
 #include <string>
+#include <unordered_map>
 
 #include "absl/log/log.h"
 #include "tensorflow/core/platform/status.h"