diff --git a/tflite/toco/graph_transformations/unfuse_activation_functions.cc b/tflite/toco/graph_transformations/unfuse_activation_functions.cc
index 00e2ba55..83778728 100644
--- a/tflite/toco/graph_transformations/unfuse_activation_functions.cc
+++ b/tflite/toco/graph_transformations/unfuse_activation_functions.cc
@@ -12,9 +12,9 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 See the License for the specific language governing permissions and
 limitations under the License.
 ==============================================================================*/
+#include <cstddef>
 #include <memory>
 #include <string>
-#include <unordered_map>
 #include <vector>
 
 #include "absl/status/status.h"
diff --git a/tflite/toco/graph_transformations/unpartition_embedding_lookup.cc b/tflite/toco/graph_transformations/unpartition_embedding_lookup.cc
index 3d7c2f1a..f8ffe70f 100644
--- a/tflite/toco/graph_transformations/unpartition_embedding_lookup.cc
+++ b/tflite/toco/graph_transformations/unpartition_embedding_lookup.cc
@@ -12,6 +12,7 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 See the License for the specific language governing permissions and
 limitations under the License.
 ==============================================================================*/
+#include <cstddef>
 #include <memory>
 #include <string>
 #include <vector>
diff --git a/tflite/toco/graph_transformations/unroll_batch_matmul.cc b/tflite/toco/graph_transformations/unroll_batch_matmul.cc
index 6a37721d..3e69608b 100644
--- a/tflite/toco/graph_transformations/unroll_batch_matmul.cc
+++ b/tflite/toco/graph_transformations/unroll_batch_matmul.cc
@@ -13,10 +13,11 @@ See the License for the specific language governing permissions and
 limitations under the License.
 ==============================================================================*/
 #include <algorithm>
+#include <cstddef>
+#include <cstdint>
+#include <iterator>
 #include <memory>
-#include <numeric>
 #include <string>
-#include <unordered_map>
 #include <vector>
 
 #include "absl/container/inlined_vector.h"