diff --git a/mindspore/ccsrc/transform/onnx/ir_exporter.cc b/mindspore/ccsrc/transform/onnx/ir_exporter.cc
index 337cf2f3d8..57c07c6889 100644
--- a/mindspore/ccsrc/transform/onnx/ir_exporter.cc
+++ b/mindspore/ccsrc/transform/onnx/ir_exporter.cc
@@ -23,8 +23,8 @@
 
 #include "ir/tensor.h"
 #include "ir/param_info.h"
-#include "debug/anf_ir_utils.h"
-#include "frontend/operator/ops.h"
+#include "ir/func_graph.h"
+#include "base/core_ops.h"
 #include "proto/onnx.pb.h"
 
 namespace mindspore {
diff --git a/mindspore/ccsrc/transform/onnx/onnx_exporter.cc b/mindspore/ccsrc/transform/onnx/onnx_exporter.cc
index 629c03849f..1d1d75573a 100644
--- a/mindspore/ccsrc/transform/onnx/onnx_exporter.cc
+++ b/mindspore/ccsrc/transform/onnx/onnx_exporter.cc
@@ -20,11 +20,11 @@
 #include <utility>
 #include <functional>
 
-#include "debug/anf_ir_utils.h"
-#include "proto/onnx.pb.h"
-#include "frontend/operator/ops.h"
 #include "ir/tensor.h"
 #include "ir/param_info.h"
+#include "ir/func_graph.h"
+#include "base/core_ops.h"
+#include "proto/onnx.pb.h"
 
 namespace mindspore {
 enum OpMergeMode {