diff --git a/mindspore/mindrecord/tools/graph_map_schema.py b/example/graph_to_mindrecord/graph_map_schema.py
similarity index 100%
rename from mindspore/mindrecord/tools/graph_map_schema.py
rename to example/graph_to_mindrecord/graph_map_schema.py
diff --git a/example/graph_to_mindrecord/writer.py b/example/graph_to_mindrecord/writer.py
index 4ada8d60ea..1024c82372 100644
--- a/example/graph_to_mindrecord/writer.py
+++ b/example/graph_to_mindrecord/writer.py
@@ -24,7 +24,7 @@ from importlib import import_module
 from multiprocessing import Pool
 
 from mindspore.mindrecord import FileWriter
-from mindspore.mindrecord import GraphMapSchema
+from graph_map_schema import GraphMapSchema
 
 
 def exec_task(task_id, parallel_writer=True):
diff --git a/mindspore/mindrecord/__init__.inter b/mindspore/mindrecord/__init__.inter
index 6fcabc2c5f..ca1d50153b 100644
--- a/mindspore/mindrecord/__init__.inter
+++ b/mindspore/mindrecord/__init__.inter
@@ -32,7 +32,6 @@ from .mindpage import MindPage
 from .shardutils import SUCCESS, FAILED
 from .tools.cifar10_to_mr import Cifar10ToMR
 from .tools.cifar100_to_mr import Cifar100ToMR
-from .tools.graph_map_schema import GraphMapSchema
 from .tools.imagenet_to_mr import ImageNetToMR
 from .tools.mnist_to_mr import MnistToMR
 
diff --git a/mindspore/mindrecord/__init__.py b/mindspore/mindrecord/__init__.py
index 44fc990399..31fb801c46 100644
--- a/mindspore/mindrecord/__init__.py
+++ b/mindspore/mindrecord/__init__.py
@@ -29,10 +29,9 @@ from .common.exceptions import *
 from .shardutils import SUCCESS, FAILED
 from .tools.cifar10_to_mr import Cifar10ToMR
 from .tools.cifar100_to_mr import Cifar100ToMR
-from .tools.graph_map_schema import GraphMapSchema
 from .tools.imagenet_to_mr import ImageNetToMR
 from .tools.mnist_to_mr import MnistToMR
 
-__all__ = ['FileWriter', 'FileReader', 'MindPage', 'GraphMapSchema',
+__all__ = ['FileWriter', 'FileReader', 'MindPage',
            'Cifar10ToMR', 'Cifar100ToMR', 'ImageNetToMR', 'MnistToMR',
            'SUCCESS', 'FAILED']