mindspore 2.0.0rc1__cp38-none-any.whl → 2.2.0__cp38-none-any.whl

mindspore/ops/bprop_mindir/TensorScatterUpdate_bprop.mindir

@@ -1,29 +0,0 @@
-
-0.1.1	MindSpore*2.0.0:�
-,get_bprop_tensor_scatter_update.1:[CNode]2:1,get_bprop_tensor_scatter_update.1:[CNode]2:1"REF::bprop.3:Default/bprop.3-op0
-�9get_bprop_tensor_scatter_update.1:tensor_scatter_update:29get_bprop_tensor_scatter_update.1:tensor_scatter_update:2"FREF::ClassType::mindspore.ops.operations.array_ops.TensorScatterUpdate:JDefault/class 'mindspore.ops.operations.array_ops.TensorScatterUpdate'-op1
-�
-get_bprop_tensor_scatter_update.1:gather_nd:6-get_bprop_tensor_scatter_update.1:gather_nd:6";REF::ClassType::mindspore.ops.operations.array_ops.GatherNd:?Default/class 'mindspore.ops.operations.array_ops.GatherNd'-op5!get_bprop_tensor_scatter_update.1*(
-&get_bprop_tensor_scatter_update.1:self*%
-#get_bprop_tensor_scatter_update.1:x*+
-)get_bprop_tensor_scatter_update.1:indices**
-(get_bprop_tensor_scatter_update.1:update*'
-%get_bprop_tensor_scatter_update.1:out*(
-&get_bprop_tensor_scatter_update.1:dout2.
-,get_bprop_tensor_scatter_update.1:[CNode]2:1:@8c2f7c478e1ce828147875b7eeb3e898896dea47389d946a79a1d2499b6915d8J/grad_array_ops.pyB�
-�

-(get_bprop_tensor_scatter_update.1:updatebprop.3:[CNode]4:3bprop.3:[CNode]4:3".REF::MetaFuncGraph::hyper_map[zeros_like_leaf]:-Default/S-Prim-hyper_map[zeros_like_leaf]-op2
-�

-&get_bprop_tensor_scatter_update.1:dout
-)get_bprop_tensor_scatter_update.1:indices
-bprop.3:[CNode]4:3bprop.3:x_grad:4bprop.3:x_grad:4">REF::get_bprop_tensor_scatter_update.1:tensor_scatter_update:2:
3
-�

-)get_bprop_tensor_scatter_update.1:indicesbprop.3:[CNode]5:5bprop.3:[CNode]5:5".REF::MetaFuncGraph::hyper_map[zeros_like_leaf]:-Default/S-Prim-hyper_map[zeros_like_leaf]-op4
-�

-&get_bprop_tensor_scatter_update.1:dout
-)get_bprop_tensor_scatter_update.1:indicesbprop.3:update_grad:7bprop.3:update_grad:7"2REF::get_bprop_tensor_scatter_update.1:gather_nd:6:
6
-�

-bprop.3:x_grad:4
-bprop.3:[CNode]5:5
-bprop.3:update_grad:7bprop.3:[CNode]6:8bprop.3:[CNode]6:8"REF::S-Prim-MakeTuple:9:Default/S-Prim-MakeTuple-op7bprop.32
-bprop.3:[CNode]6:8P
b&
-S-Prim-MakeTuple:9S-Prim-MakeTupleh

mindspore/ops/bprop_mindir/TensorShape_bprop.mindir

@@ -1,14 +0,0 @@
-
-0.1.1	MindSpore*2.0.0:�
-r$get_bprop_tensorshape.21:[CNode]22:1$get_bprop_tensorshape.21:[CNode]22:1"
-get_bprop_tensorshape.21:self*
-get_bprop_tensorshape.21:x*
-get_bprop_tensorshape.21:out*
-get_bprop_tensorshape.21:dout2&
-$get_bprop_tensorshape.21:[CNode]22:1:@62dc7ba53f71834ce824222195cb6a9af0b1f9d5c7e86cd3ac9eecfb0e3e1b3dJ/grad_array_ops.pyB�
-�

-get_bprop_tensorshape.21:xbprop.23:[CNode]24:2bprop.23:[CNode]24:2".REF::MetaFuncGraph::hyper_map[zeros_like_leaf]:.Default/S-Prim-hyper_map[zeros_like_leaf]-op14
-z
-bprop.23:[CNode]24:2bprop.23:[CNode]25:3bprop.23:[CNode]25:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op15bprop.232
-bprop.23:[CNode]25:3P
b&
-S-Prim-MakeTuple:4S-Prim-MakeTupleh

mindspore/ops/bprop_mindir/TransShape_bprop.mindir

@@ -1,23 +0,0 @@
-
-0.1.1	MindSpore*2.0.0:�
-k"get_bprop_trans_shape.1:[CNode]2:1"get_bprop_trans_shape.1:[CNode]2:1"REF::bprop.3:Default/bprop.3-op0
-�
get_bprop_trans_shape.1:op:2get_bprop_trans_shape.1:op:2"=REF::ClassType::mindspore.ops.operations.array_ops.TransShape:ADefault/class 'mindspore.ops.operations.array_ops.TransShape'-op1get_bprop_trans_shape.1*
-get_bprop_trans_shape.1:self*
-get_bprop_trans_shape.1:x*
-get_bprop_trans_shape.1:shape*
-get_bprop_trans_shape.1:out*
-get_bprop_trans_shape.1:dout2$
-"get_bprop_trans_shape.1:[CNode]2:1:@adaf31d907574c3f63f9202dc6a36f2f70f0e7caf15e7294fa4a4f2427bc8d02J/grad_array_ops.pyB�
-r
-get_bprop_trans_shape.1:xbprop.3:[CNode]4:3bprop.3:[CNode]4:3"REF::S-Prim-Shape:4:Default/S-Prim-Shape-op2
-t
-get_bprop_trans_shape.1:dout
-bprop.3:[CNode]4:3bprop.3:dx:5bprop.3:dx:5"!REF::get_bprop_trans_shape.1:op:2:
3
-�

-get_bprop_trans_shape.1:shapebprop.3:[CNode]5:6bprop.3:[CNode]5:6".REF::MetaFuncGraph::hyper_map[zeros_like_leaf]:-Default/S-Prim-hyper_map[zeros_like_leaf]-op4
-�

-bprop.3:dx:5
-bprop.3:[CNode]5:6bprop.3:[CNode]6:7bprop.3:[CNode]6:7"REF::S-Prim-MakeTuple:8:Default/S-Prim-MakeTuple-op5bprop.32
-bprop.3:[CNode]6:7P
b
-S-Prim-Shape:4S-Prim-Shapeb&
-S-Prim-MakeTuple:8S-Prim-MakeTupleh

mindspore/ops/bprop_mindir/TruncateDiv_bprop.mindir

@@ -1,19 +0,0 @@
-
-0.1.1	MindSpore*2.0.0:�
-�
)get_bprop_truncate_div.1125:[CNode]1126:1)get_bprop_truncate_div.1125:[CNode]1126:1"REF::bprop.1127:Default/bprop.1127-op886get_bprop_truncate_div.1125*"
- get_bprop_truncate_div.1125:self*
-get_bprop_truncate_div.1125:x*
-get_bprop_truncate_div.1125:y*!
-get_bprop_truncate_div.1125:out*"
- get_bprop_truncate_div.1125:dout2+
-)get_bprop_truncate_div.1125:[CNode]1126:1:@a8f96e783e49fe455f7fee2dbda70043761c9bb3cbcd1c9ad3cd37eecd1f80e5J/grad_math_ops.pyB�
-�

-get_bprop_truncate_div.1125:xbprop.1127:[CNode]1128:2bprop.1127:[CNode]1128:2".REF::MetaFuncGraph::hyper_map[zeros_like_leaf]:/Default/S-Prim-hyper_map[zeros_like_leaf]-op887
-�

-get_bprop_truncate_div.1125:ybprop.1127:[CNode]1129:3bprop.1127:[CNode]1129:3".REF::MetaFuncGraph::hyper_map[zeros_like_leaf]:/Default/S-Prim-hyper_map[zeros_like_leaf]-op888
-�

-bprop.1127:[CNode]1128:2
-bprop.1127:[CNode]1129:3bprop.1127:[CNode]1130:4bprop.1127:[CNode]1130:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op889
-bprop.11272
-bprop.1127:[CNode]1130:4P
b&
-S-Prim-MakeTuple:5S-Prim-MakeTupleh

mindspore/ops/bprop_mindir/TupleGetItem_bprop.mindir

@@ -1,20 +0,0 @@
-
-0.1.1	MindSpore*2.0.0:�
-�

-bprop_tuple_getitem.1:data bprop_tuple_getitem.1:[CNode]2:1 bprop_tuple_getitem.1:[CNode]2:1".REF::MetaFuncGraph::hyper_map[zeros_like_leaf]:-Default/S-Prim-hyper_map[zeros_like_leaf]-op0
-�

- bprop_tuple_getitem.1:[CNode]2:1
-bprop_tuple_getitem.1:idx
-bprop_tuple_getitem.1:dout bprop_tuple_getitem.1:[CNode]3:2 bprop_tuple_getitem.1:[CNode]3:2"REF::S-Prim-tuple_setitem:3: Default/S-Prim-tuple_setitem-op1
-�

-bprop_tuple_getitem.1:idx bprop_tuple_getitem.1:[CNode]4:4 bprop_tuple_getitem.1:[CNode]4:4".REF::MetaFuncGraph::hyper_map[zeros_like_leaf]:-Default/S-Prim-hyper_map[zeros_like_leaf]-op2
-�

- bprop_tuple_getitem.1:[CNode]3:2
- bprop_tuple_getitem.1:[CNode]4:4 bprop_tuple_getitem.1:[CNode]5:5 bprop_tuple_getitem.1:[CNode]5:5"REF::S-Prim-MakeTuple:6:Default/S-Prim-MakeTuple-op3bprop_tuple_getitem.1*
-bprop_tuple_getitem.1:data*
-bprop_tuple_getitem.1:idx*
-bprop_tuple_getitem.1:out*
-bprop_tuple_getitem.1:dout2"
- bprop_tuple_getitem.1:[CNode]5:5:@badb8d2f379272a9bfe28f5e50b98c3de21620b357eb5d206a8abc1d48964179J/grad_implementations.pyP
b.
-S-Prim-tuple_setitem:3S-Prim-tuple_setitemb&
-S-Prim-MakeTuple:6S-Prim-MakeTupleh

mindspore/ops/bprop_mindir/Unique_bprop.mindir

@@ -1,16 +0,0 @@
-
-0.1.1	MindSpore*2.0.0:�
-aget_bprop_unique.1:[CNode]2:1get_bprop_unique.1:[CNode]2:1"REF::bprop.3:Default/bprop.3-op0
-�
get_bprop_unique.1:op:2get_bprop_unique.1:op:2"=REF::ClassType::mindspore.ops.operations._grad_ops.UniqueGrad:ADefault/class 'mindspore.ops.operations._grad_ops.UniqueGrad'-op1get_bprop_unique.1*
-get_bprop_unique.1:self*
-get_bprop_unique.1:x*
-get_bprop_unique.1:out*
-get_bprop_unique.1:dout2
-get_bprop_unique.1:[CNode]2:1:@250b2c34ad30cb5948758d14a57231a58534cf4e271d6445ab7a48ae067de84dJ/grad_array_ops.pyB�

-n
-get_bprop_unique.1:dout
-get_bprop_unique.1:outbprop.3:dx:3bprop.3:dx:3"REF::get_bprop_unique.1:op:2:
2
-m
-bprop.3:dx:3bprop.3:[CNode]4:4bprop.3:[CNode]4:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op3bprop.32
-bprop.3:[CNode]4:4P
b&
-S-Prim-MakeTuple:5S-Prim-MakeTupleh

mindspore/ops/bprop_mindir/Unstack_bprop.mindir

@@ -1,22 +0,0 @@
-
-0.1.1	MindSpore*2.0.0:�
-q"get_bprop_unstack.924:[CNode]925:1"get_bprop_unstack.924:[CNode]925:1"REF::bprop.926:Default/bprop.926-op749
-<[ValueNode]927[ValueNode]927"Constant*
-value*axis�

-�

-get_bprop_unstack.924:self
-[ValueNode]927get_bprop_unstack.924:axis:2get_bprop_unstack.924:axis:2"REF::getattr:3:Default/getattr-op750get_bprop_unstack.924*
-get_bprop_unstack.924:self*
-get_bprop_unstack.924:x*
-get_bprop_unstack.924:out*
-get_bprop_unstack.924:dout2$
-"get_bprop_unstack.924:[CNode]925:1:@c4c274247cc015c77285a377298c8228ea58370ed44389f89a8ac0e06f9aec24J/grad_array_ops.pyB�
-�

-get_bprop_unstack.924:axis:2bprop.926:unstack_grad:4bprop.926:unstack_grad:4"8REF::ClassType::mindspore.ops.operations.array_ops.Stack:>Default/class 'mindspore.ops.operations.array_ops.Stack'-op751
-b
-get_bprop_unstack.924:doutbprop.926:out:5bprop.926:out:5"REF::bprop.926:unstack_grad:4:752
-z
-bprop.926:out:5bprop.926:[CNode]928:6bprop.926:[CNode]928:6"REF::S-Prim-MakeTuple:7:Default/S-Prim-MakeTuple-op753	bprop.9262
-bprop.926:[CNode]928:6P
b
-	getattr:3getattrb&
-S-Prim-MakeTuple:7S-Prim-MakeTupleh

mindspore/ops/bprop_mindir/UpsampleNearest3D_bprop.mindir

@@ -1,32 +0,0 @@
-
-0.1.1	MindSpore*2.0.0:�	
-�
0get_bprop_upsample_nearest_3d_grad.2:[CNode]13:10get_bprop_upsample_nearest_3d_grad.2:[CNode]13:1"REF::bprop.4:Default/bprop.4-op0
-�
0get_bprop_upsample_nearest_3d_grad.2:get_shape:20get_bprop_upsample_nearest_3d_grad.2:get_shape:2"8REF::ClassType::mindspore.ops.operations.array_ops.Shape:LDefault/MindIRClassType:class 'mindspore.ops.operations.array_ops.Shape'-op1
-A
-value*output_size�

-�

-)get_bprop_upsample_nearest_3d_grad.2:self
-
-<
-value*scales�

-�

-)get_bprop_upsample_nearest_3d_grad.2:self
-
-)get_bprop_upsample_nearest_3d_grad.2:self*.
-,get_bprop_upsample_nearest_3d_grad.2:input_x**
-(get_bprop_upsample_nearest_3d_grad.2:out*+
-)get_bprop_upsample_nearest_3d_grad.2:dout22
-0get_bprop_upsample_nearest_3d_grad.2:[CNode]13:1:@89f303661abc4ca7469d2cbbfdc77c210c2a3053877fbd2b8ab19c9b86f6f56aJ/grad_nn_ops.pyB�
-�

-,get_bprop_upsample_nearest_3d_grad.2:input_xbprop.4:[CNode]1:3bprop.4:[CNode]1:3"5REF::get_bprop_upsample_nearest_3d_grad.2:get_shape:2:
2
-�
-bprop.4:[CNode]1:3
-2get_bprop_upsample_nearest_3d_grad.2:output_size:4
--get_bprop_upsample_nearest_3d_grad.2:scales:6bprop.4:input_grad:7bprop.4:input_grad:7"HREF::ClassType::mindspore.ops.operations._grad_ops.UpsampleNearest3DGrad:\Default/MindIRClassType:class 'mindspore.ops.operations._grad_ops.UpsampleNearest3DGrad'-op5
-e
-)get_bprop_upsample_nearest_3d_grad.2:doutbprop.4:dx:8bprop.4:dx:8"REF::bprop.4:input_grad:7:
6
-n
-bprop.4:dx:8bprop.4:[CNode]3:9bprop.4:[CNode]3:9"REF::S-Prim-MakeTuple:10:Default/S-Prim-MakeTuple-op7bprop.42
-bprop.4:[CNode]3:9P
b
-	getattr:5getattrb'
-S-Prim-MakeTuple:10S-Prim-MakeTupleh

mindspore/ops/bprop_mindir/UpsampleTrilinear3D_bprop.mindir

@@ -1,38 +0,0 @@
-
-0.1.1	MindSpore*2.0.0:�
-�
2get_bprop_upsample_trilinear_3d_grad.2:[CNode]13:12get_bprop_upsample_trilinear_3d_grad.2:[CNode]13:1"REF::bprop.4:Default/bprop.4-op0
-�
2get_bprop_upsample_trilinear_3d_grad.2:get_shape:22get_bprop_upsample_trilinear_3d_grad.2:get_shape:2"8REF::ClassType::mindspore.ops.operations.array_ops.Shape:LDefault/MindIRClassType:class 'mindspore.ops.operations.array_ops.Shape'-op1
-A
-value*output_size�

-�

-+get_bprop_upsample_trilinear_3d_grad.2:self
-
-<
-value*scales�

-�

-+get_bprop_upsample_trilinear_3d_grad.2:self
-
-C
-value*
-�

-+get_bprop_upsample_trilinear_3d_grad.2:self
-
-+get_bprop_upsample_trilinear_3d_grad.2:self*0
-.get_bprop_upsample_trilinear_3d_grad.2:input_x*,
-*get_bprop_upsample_trilinear_3d_grad.2:out*-
-+get_bprop_upsample_trilinear_3d_grad.2:dout24
-2get_bprop_upsample_trilinear_3d_grad.2:[CNode]13:1:@de557b6ab60e23e396148159c8aa5c5a3a0a77cf25b1b19f25b6e067c283d1e4J/grad_nn_ops.pyB�
-�

-.get_bprop_upsample_trilinear_3d_grad.2:input_xbprop.4:[CNode]1:3bprop.4:[CNode]1:3"7REF::get_bprop_upsample_trilinear_3d_grad.2:get_shape:2:
2
-�
-bprop.4:[CNode]1:3
-4get_bprop_upsample_trilinear_3d_grad.2:output_size:4
-/get_bprop_upsample_trilinear_3d_grad.2:scales:6
-6get_bprop_upsample_trilinear_3d_grad.2:align_corners:7bprop.4:input_grad:8bprop.4:input_grad:8"JREF::ClassType::mindspore.ops.operations._grad_ops.UpsampleTrilinear3DGrad:^Default/MindIRClassType:class 'mindspore.ops.operations._grad_ops.UpsampleTrilinear3DGrad'-op6
-g
-+get_bprop_upsample_trilinear_3d_grad.2:doutbprop.4:dx:9bprop.4:dx:9"REF::bprop.4:input_grad:8:
7
-p
-bprop.4:dx:9bprop.4:[CNode]3:10bprop.4:[CNode]3:10"REF::S-Prim-MakeTuple:11:Default/S-Prim-MakeTuple-op8bprop.42
-bprop.4:[CNode]3:10P
b
-	getattr:5getattrb'
-S-Prim-MakeTuple:11S-Prim-MakeTupleh

mindspore/ops/bprop_mindir/ZerosLike_bprop.mindir

@@ -1,15 +0,0 @@
-
-0.1.1	MindSpore*2.0.0:�
-{&get_bprop_zeroslike.1001:[CNode]1002:1&get_bprop_zeroslike.1001:[CNode]1002:1"REF::bprop.1003:Default/bprop.1003-op802get_bprop_zeroslike.1001*
-get_bprop_zeroslike.1001:self*
-get_bprop_zeroslike.1001:x*
-get_bprop_zeroslike.1001:out*
-get_bprop_zeroslike.1001:dout2(
-&get_bprop_zeroslike.1001:[CNode]1002:1:@019bc2e3d6af94c73e4507b14158fc13a846c43d6f52e4d09d3044897de5461fJ/grad_array_ops.pyB�
-�

-get_bprop_zeroslike.1001:xbprop.1003:[CNode]1004:2bprop.1003:[CNode]1004:2".REF::MetaFuncGraph::hyper_map[zeros_like_leaf]:/Default/S-Prim-hyper_map[zeros_like_leaf]-op803
-�

-bprop.1003:[CNode]1004:2bprop.1003:[CNode]1005:3bprop.1003:[CNode]1005:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op804
-bprop.10032
-bprop.1003:[CNode]1005:3P
b&
-S-Prim-MakeTuple:4S-Prim-MakeTupleh

mindspore/ops/bprop_mindir/generate_mindir.py

@@ -1,114 +0,0 @@
-# Copyright 2021-2022 Huawei Technologies Co., Ltd
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# 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.
-# ============================================================================
-"""Generate the mindir for bprop"""
-from __future__ import absolute_import
-import os
-import shutil
-import argparse
-import logging
-
-import mindspore.ops._grad as g
-from mindspore._c_expression import _export_bprop_mindir
-from mindspore.ops._grad.grad_base import bprop_getters, bprops
-
-logging.getLogger().setLevel(logging.INFO)
-os.environ['MS_DEV_EXPORT_BPROP_MINDIR'] = '1'
-
-
-def run_generate(bprop_mindir_install_dir, bprop_map, force_update):
-    """Run the generate process of bprop mindir."""
-    for op_name in bprop_map.keys():
-        if not isinstance(op_name, str):
-            continue
-        if os.path.isfile(os.path.join(bprop_mindir_install_dir, op_name + "_bprop.mindir")):
-            _export_bprop_mindir(op_name, force_update)
-
-
-def run_generate_with_op_name(op_name, force_update):
-    """Run the generate process of bprop mindir for single op."""
-    _export_bprop_mindir(op_name, force_update)
-
-
-if __name__ == "__main__":
-    parser = argparse.ArgumentParser(description="Bprop mindir generator")
-    parser.add_argument('--mindspore_path', type=str, default=None,
-                        help="The absolute path of the mindspore root directory where the bprop source files has been \
-                        modified. If not specified, it will find the bprop source files in your mindspore installed \
-                        path. Default: None.")
-    parser.add_argument('--op', type=str, default=None,
-                        help="The name of the operator whose bprop is to be transformed to mindir file. If not \
-                        specified, it will generate all the mindir files of bprop which has already been transform \
-                        to mindir. Default: None.")
-    parser.add_argument('--force', type=bool, default=False,
-                        help="Whether to force to generate mindir file of a bprop function regardless of nothing \
-                        changed. Default: False.")
-
-    args_opt = parser.parse_args()
-    # mindspore/ops/_grad/__init__.py
-    BPROP_PATH = g.__file__
-    bprop_installed_dir = BPROP_PATH[: BPROP_PATH.rindex('/')]
-    bprop_mindir_export_dir = os.path.join(bprop_installed_dir, "..", "bprop_mindir")
-
-    mindspore_path = args_opt.mindspore_path
-    bprop_src_dir = None
-    bprop_mindir_src_dir = None
-    if mindspore_path is not None:
-        mindspore_path = mindspore_path.rstrip('/')
-        python_ops_dir = os.path.join(mindspore_path, "mindspore", "python", "mindspore", "ops")
-        bprop_src_dir = os.path.join(python_ops_dir, "_grad")
-        bprop_mindir_src_dir = os.path.join(python_ops_dir, "bprop_mindir")
-
-    copy_flag = bprop_src_dir is not None and bprop_src_dir != bprop_installed_dir
-    # If the specified bprop source directory is not on the mindspore installed path,
-    # copy the bprop source files to the installed path.
-    BACKUP_SUFFIX = "_generate_bak"
-    if copy_flag:
-        shutil.rmtree(bprop_installed_dir + BACKUP_SUFFIX, ignore_errors=True)
-        os.rename(bprop_installed_dir, bprop_installed_dir + BACKUP_SUFFIX)
-        os.mkdir(bprop_installed_dir)
-        ls = os.listdir(bprop_src_dir)
-        for line in ls:
-            file_path = os.path.join(bprop_src_dir, line)
-            if os.path.isfile(file_path):
-                shutil.copy(file_path, bprop_installed_dir)
-                logging.info("copied: %s", file_path)
-
-    force = args_opt.force
-    op = args_opt.op
-    if op is None:
-        run_generate(bprop_mindir_export_dir, bprop_getters, force)
-        run_generate(bprop_mindir_export_dir, bprops, force)
-    else:
-        run_generate_with_op_name(op, force)
-
-    # If the specified bprop source directory is not on the mindspore installed path,
-    # copy the generated mindir files to the mindir directory relative to the specified path.
-    if copy_flag:
-        shutil.rmtree(bprop_installed_dir)
-        os.rename(bprop_installed_dir + BACKUP_SUFFIX, bprop_installed_dir)
-        ls = os.listdir(bprop_mindir_export_dir)
-        for line in ls:
-            file_path = os.path.join(bprop_mindir_export_dir, line)
-            if file_path.endswith(".mindir") and os.path.isfile(file_path):
-                os.chmod(file_path, 0o664)
-                shutil.copy(file_path, bprop_mindir_src_dir)
-                logging.info("copied: %s", file_path)
-
-        logging.info("The new bprop mindir files has been generated in the path \"%s\"", bprop_mindir_src_dir)
-    else:
-        logging.info("The new bprop mindir files has been generated in the path \"%s\", "
-                     "copy the *.mindir to your mindspore path or PYTHONPATH if necessary.", bprop_mindir_export_dir)
-
-    del os.environ['MS_DEV_EXPORT_BPROP_MINDIR']

mindspore/rewrite/node_visitor.py

@@ -1,44 +0,0 @@
-# Copyright 2022 Huawei Technologies Co., Ltd
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# 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.
-# ============================================================================
-"""Visit nods of SymbolTree."""
-
-
-class NodeVisitor:
-    """Iterator class to access SymbolTree nodes"""
-    def __init__(self, stree):
-        self._stree = stree
-        self._nodes = []
-        self._index = 0
-
-    def __iter__(self):
-        self._nodes = list(self._stree.get_nodes_dict().values())
-        self._index = 0
-        return self
-
-    def __next__(self):
-        if self._index < len(self._nodes):
-            node = self._nodes[self._index]
-            self._index += 1
-            return node
-
-        raise StopIteration
-
-    def append_node(self, node):
-        """append new node to iterator"""
-        self._nodes.append(node)
-
-    def remove_node(self, node):
-        """remove node of iterator"""
-        self._nodes.remove(node)

mindspore/rewrite/topological_manager.py

@@ -1,203 +0,0 @@
-# Copyright 2022 Huawei Technologies Co., Ltd
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# 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.
-# ============================================================================
-"""SymbolTree topological-relationship manager."""
-from typing import Tuple
-
-from .api.scoped_value import ScopedValue
-from .node import Node
-from .common.observable import Observable
-from .common.event import Event
-
-
-class TopoManager(Observable):
-    """SymbolTree topological-relationship manager."""
-
-    def __init__(self):
-        """
-        Constructor of TopoManager.
-        Init provider and consumer.
-        Key of dict is an instance of ScopedValue.
-        Value of dict is a tuple whose first is an instance of Node, whose second is an index.
-        It means node's index th arg is argument
-        """
-        super().__init__()
-        self._target_provider: {ScopedValue, (Node, int)} = {}
-        self._target_consumer: {ScopedValue, [(Node, int)]} = {}
-
-    def get_node_users(self, node: Node) -> [Tuple[Node, int]]:
-        """
-        Get all nodes which depend on node corresponding to node_or_name.
-
-        Args:
-            node (Node): An instance of node.
-
-        Returns:
-            A list of nodes represents node users.
-        """
-        targets = node.get_targets()
-        results = []
-        for target in targets:
-            consumers = self._target_consumer.get(target)
-            if consumers is None:
-                continue
-            results.extend(consumers)
-        unique_results = []
-        for result in results:
-            if result not in unique_results:
-                unique_results.append(result)
-        return unique_results
-
-    def topo_changed(self):
-        """
-        The function is executed when an Event.TopologicalChangeEvent event is received.
-        """
-        self.changed(Event.TopologicalChangeEvent)
-
-    def _add_consumer(self, product: ScopedValue, consumer: Node, index):
-        """
-        Add a consumer to consumer dict.
-
-        Args:
-            product (ScopedValue): An instance of ScopedValue represents product to be consumed.
-            consumer (Node): An instance of Node represents consumer.
-            index (int): A int represents which input of consumer is the product.
-        """
-        consumers = self._target_consumer.get(product)
-        if consumers is None:
-            self._target_consumer[product] = [(consumer, index)]
-        else:
-            self._target_consumer.get(product).append((consumer, index))
-
-    def _erase_provider(self, product: ScopedValue):
-        """
-        Erase a provider from provider dict.
-
-        Args:
-            product (ScopedValue): An instance of ScopedValue represents product to be erased.
-        """
-        if self._target_provider.get(product) is not None:
-            self._target_provider.pop(product)
-
-    def _erase_consumer(self, product: ScopedValue, consumer: Node):
-        """
-        Erase a consumer from consumer dict.
-
-        Args:
-            product (ScopedValue): An instance of ScopedValue represents product whose consumer would be erased.
-            consumer (Node): An instance of Node which would be erased as a consumer.
-        """
-        consumers = self._target_consumer.get(product)
-        if consumers is None:
-            return
-        for i in range(len(consumers) - 1, -1, -1):
-            exist_ele = consumers[i]
-            if id(exist_ele[0]) == id(consumer):
-                consumers.pop(i)
-
-    def _update_node_inputs(self, node: Node) -> [Node]:
-        """
-        Update inputs of node by current provider dict and consumer dict.
-
-        Args:
-            node (Node): An instance of Node whose inputs will be updated.
-
-        Returns:
-            A list of instance of nodes represents inputs of node.
-        """
-        if node.get_normalized_args() is None:
-            node.set_inputs([])
-            return []
-        inputs = []
-        for arg in node.get_normalized_args().values():
-            provider = self._target_provider.get(arg)
-            # some arg of some node may be self.xxx which is not an output of another node
-            if provider is not None:
-                inputs.append(provider[0])
-        node.set_inputs(inputs)
-        return inputs
-
-    def on_insert_node(self, node: Node):
-        """
-        Update provider dict and consumer dict while inserting node into SymbolTree and update inputs of node by updated
-        provider dict and consumer dict.
-
-        Args:
-            node (Node): An instance of Node which been inserted into SymbolTree.
-        """
-        if node.get_targets() is not None:
-            for i in range(0, len(node.get_targets())):
-                target = node.get_targets()[i]
-                if target.value == "_":
-                    continue
-                if self._target_provider.get(target) is not None:
-                    raise RuntimeError("target duplicated:", target)
-                self._target_provider[target] = (node, i)
-        if node.get_normalized_args() is not None:
-            for index, arg in enumerate(node.get_normalized_args().values()):
-                self._add_consumer(arg, node, index)
-        self._update_node_inputs(node)
-        self.topo_changed()
-
-    def on_erase_node(self, node: Node):
-        """
-        Update provider dict and consumer dict while erasing node from SymbolTree.
-
-        Args:
-            node (Node): An instance of Node which been erased from SymbolTree.
-        """
-        if node.get_targets() is not None:
-            for target in node.get_targets():
-                consumers = self._target_consumer.get(target)
-                if consumers is not None and consumers:
-                    raise RuntimeError("Only support erase isolated node: ", node.get_name(), target)
-                self._erase_provider(target)
-        if node.get_normalized_args() is not None:
-            for arg in node.get_normalized_args().values():
-                self._erase_consumer(arg, node)
-        # clear inputs of node rather than call _update_node_inputs because node is already erase from consumer dict
-        node.set_inputs([])
-        self.topo_changed()
-
-    def on_update_arg(self, node: Node, arg_idx: int, old_arg: ScopedValue, new_arg: ScopedValue):
-        """
-        Update provider dict and consumer dict while updating argument of node and update inputs of node by updated
-        provider dict and consumer dict.
-
-        Args:
-            node (Node): An instance of Node whose arguments being updated.
-            arg_idx (int): An int indicates which argument of node being updated.
-            old_arg (ScopedValue): An instance of ScopedValue represents original argument.
-            new_arg (ScopedValue): An instance of ScopedValue represents new argument.
-        """
-        self._erase_consumer(old_arg, node)
-        self._add_consumer(new_arg, node, arg_idx)
-        self._update_node_inputs(node)
-        self.topo_changed()
-
-    def dump(self, title=""):
-        """
-        Dump topological relation.
-
-        Args:
-            title (str): A string as a title will be printed before dumping topological relation.
-        """
-        print(f"{title}------------------------------------------------------------------------------------")
-        for k, v in self._target_provider.items():
-            print(f"{v[0].get_name()} produces {k.value}")
-        for k, v in self._target_consumer.items():
-            print(f"{k.value} is consumed by: ")
-            for ele in v:
-                print(ele[0].get_name())
-        print(f"-----------------------------------------------------------------------------------------")