Automated Code Change

PiperOrigin-RevId: 706662854
google-ai-edge · Dec 16, 2024 · 987c536 · 987c536
1 parent c16e477
commit 987c536
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/tflite/tools/signature/signature_def_util.cc b/tflite/tools/signature/signature_def_util.cc
@@ -14,6 +14,8 @@ limitations under the License.
 ==============================================================================*/
 #include "tflite/tools/signature/signature_def_util.h"
 
+#include <cstddef>
+#include <cstdint>
 #include <map>
 #include <memory>
 #include <string>

diff --git a/tflite/tools/signature/signature_def_util.h b/tflite/tools/signature/signature_def_util.h
@@ -15,8 +15,10 @@ limitations under the License.
 #ifndef TENSORFLOW_LITE_TOOLS_SIGNATURE_SIGNATURE_DEF_UTIL_H_
 #define TENSORFLOW_LITE_TOOLS_SIGNATURE_SIGNATURE_DEF_UTIL_H_
 
+#include <map>
 #include <string>
 
+#include "absl/status/status.h"
 #include "tensorflow/core/lib/core/status.h"
 #include "tensorflow/core/platform/status.h"
 #include "tensorflow/core/protobuf/meta_graph.pb.h"

diff --git a/tflite/tools/signature/signature_def_util_wrapper_pybind11.cc b/tflite/tools/signature/signature_def_util_wrapper_pybind11.cc
@@ -12,7 +12,11 @@ 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 <cstdint>
+#include <map>
+#include <stdexcept>
 #include <string>
+#include <vector>
 
 #include "absl/status/status.h"
 #include "pybind11/pybind11.h"  // from @pybind11