--- a/tensorflow/core/kernels/sparse_tensors_map_ops.cc	2021-08-05 09:54:10.591394528 +0800
+++ b/tensorflow/core/kernels/sparse_tensors_map_ops.cc	2021-08-05 12:22:51.148694390 +0800
@@ -21,6 +21,8 @@
 #include <utility>
 #include <vector>
 
+#include "tensorflow/core/framework/op_kernel.h"
+#include "tensorflow/core/framework/register_types.h"
 #include "tensorflow/core/framework/resource_mgr.h"
 #include "tensorflow/core/framework/tensor.h"
 #include "tensorflow/core/framework/tensor_util.h"