Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
magicwindyyd
mindspore
提交
1a54785f
M
mindspore
项目概览
magicwindyyd
/
mindspore
与 Fork 源项目一致
Fork自
MindSpore / mindspore
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
mindspore
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
1a54785f
编写于
8月 25, 2020
作者:
P
panyifeng
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
remove name arg from gradoperation
上级
b5ed5466
变更
187
展开全部
隐藏空白更改
内联
并排
Showing
187 changed file
with
243 addition
and
269 deletion
+243
-269
mindspore/nn/wrap/cell_wrapper.py
mindspore/nn/wrap/cell_wrapper.py
+2
-2
mindspore/nn/wrap/grad_reducer.py
mindspore/nn/wrap/grad_reducer.py
+1
-1
mindspore/nn/wrap/loss_scale.py
mindspore/nn/wrap/loss_scale.py
+1
-1
mindspore/ops/composite/base.py
mindspore/ops/composite/base.py
+3
-4
mindspore/ops/operations/debug_ops.py
mindspore/ops/operations/debug_ops.py
+2
-2
model_zoo/official/cv/faster_rcnn/src/network_define.py
model_zoo/official/cv/faster_rcnn/src/network_define.py
+1
-2
model_zoo/official/cv/maskrcnn/src/network_define.py
model_zoo/official/cv/maskrcnn/src/network_define.py
+1
-2
model_zoo/official/cv/resnet_thor/src/grad_reducer_thor.py
model_zoo/official/cv/resnet_thor/src/grad_reducer_thor.py
+1
-1
model_zoo/official/cv/ssd/src/ssd.py
model_zoo/official/cv/ssd/src/ssd.py
+1
-1
model_zoo/official/cv/warpctc/src/warpctc_for_train.py
model_zoo/official/cv/warpctc/src/warpctc_for_train.py
+1
-1
model_zoo/official/cv/yolov3_darknet53/src/yolo.py
model_zoo/official/cv/yolov3_darknet53/src/yolo.py
+1
-1
model_zoo/official/cv/yolov3_darknet53_quant/src/yolo.py
model_zoo/official/cv/yolov3_darknet53_quant/src/yolo.py
+1
-1
model_zoo/official/cv/yolov3_resnet18/src/yolov3.py
model_zoo/official/cv/yolov3_resnet18/src/yolov3.py
+1
-1
model_zoo/official/gnn/gat/src/utils.py
model_zoo/official/gnn/gat/src/utils.py
+1
-1
model_zoo/official/gnn/gcn/src/metrics.py
model_zoo/official/gnn/gcn/src/metrics.py
+1
-1
model_zoo/official/nlp/bert/src/bert_for_finetune.py
model_zoo/official/nlp/bert/src/bert_for_finetune.py
+2
-3
model_zoo/official/nlp/bert/src/bert_for_pre_training.py
model_zoo/official/nlp/bert/src/bert_for_pre_training.py
+2
-3
model_zoo/official/nlp/bert_thor/src/bert_for_pre_training.py
...l_zoo/official/nlp/bert_thor/src/bert_for_pre_training.py
+2
-3
model_zoo/official/nlp/bert_thor/src/grad_reducer_thor.py
model_zoo/official/nlp/bert_thor/src/grad_reducer_thor.py
+1
-1
model_zoo/official/nlp/mass/src/transformer/transformer_for_train.py
...fficial/nlp/mass/src/transformer/transformer_for_train.py
+1
-1
model_zoo/official/nlp/tinybert/src/tinybert_for_gd_td.py
model_zoo/official/nlp/tinybert/src/tinybert_for_gd_td.py
+4
-8
model_zoo/official/nlp/transformer/src/transformer_for_train.py
...zoo/official/nlp/transformer/src/transformer_for_train.py
+2
-3
model_zoo/official/recommend/deepfm/src/deepfm.py
model_zoo/official/recommend/deepfm/src/deepfm.py
+1
-1
model_zoo/official/recommend/wide_and_deep/src/wide_and_deep.py
...zoo/official/recommend/wide_and_deep/src/wide_and_deep.py
+2
-2
model_zoo/official/recommend/wide_and_deep_multitable/src/wide_and_deep.py
...l/recommend/wide_and_deep_multitable/src/wide_and_deep.py
+2
-4
tests/mindspore_test_framework/components/function/compile_gradient_wrt_inputs.py
...mework/components/function/compile_gradient_wrt_inputs.py
+1
-1
tests/mindspore_test_framework/components/function/compile_gradient_wrt_params.py
...mework/components/function/compile_gradient_wrt_params.py
+1
-1
tests/mindspore_test_framework/components/function/init_params_with_rand_and_run_gradient_wrt_inputs.py
...tion/init_params_with_rand_and_run_gradient_wrt_inputs.py
+1
-1
tests/mindspore_test_framework/components/function/init_params_with_rand_and_run_gradient_wrt_params.py
...tion/init_params_with_rand_and_run_gradient_wrt_params.py
+1
-1
tests/mindspore_test_framework/components/function/run_gradient_wrt_inputs.py
..._framework/components/function/run_gradient_wrt_inputs.py
+1
-1
tests/mindspore_test_framework/components/function/run_gradient_wrt_params.py
..._framework/components/function/run_gradient_wrt_params.py
+1
-1
tests/mindspore_test_framework/utils/block_util.py
tests/mindspore_test_framework/utils/block_util.py
+1
-1
tests/mindspore_test_framework/utils/bprop_util.py
tests/mindspore_test_framework/utils/bprop_util.py
+1
-1
tests/mindspore_test_framework/utils/check_gradient.py
tests/mindspore_test_framework/utils/check_gradient.py
+3
-3
tests/ops_common.py
tests/ops_common.py
+1
-1
tests/perf_test/test_lenet.py
tests/perf_test/test_lenet.py
+1
-1
tests/st/control/test_cont_grad.py
tests/st/control/test_cont_grad.py
+2
-2
tests/st/gnn/test_gnn_aggregator.py
tests/st/gnn/test_gnn_aggregator.py
+1
-1
tests/st/model_zoo_tests/wide_and_deep/python_file_for_ci/wide_and_deep.py
...o_tests/wide_and_deep/python_file_for_ci/wide_and_deep.py
+2
-2
tests/st/model_zoo_tests/yolov3/src/yolov3.py
tests/st/model_zoo_tests/yolov3/src/yolov3.py
+1
-1
tests/st/networks/models/bert/src/bert_for_pre_training.py
tests/st/networks/models/bert/src/bert_for_pre_training.py
+2
-3
tests/st/networks/models/bert/src/utils.py
tests/st/networks/models/bert/src/utils.py
+1
-2
tests/st/networks/models/resnet50/src_thor/grad_reducer_thor.py
...st/networks/models/resnet50/src_thor/grad_reducer_thor.py
+1
-1
tests/st/networks/test_cell_bprop.py
tests/st/networks/test_cell_bprop.py
+2
-2
tests/st/ops/ascend/test_addn.py
tests/st/ops/ascend/test_addn.py
+1
-1
tests/st/ops/ascend/test_conv_grad.py
tests/st/ops/ascend/test_conv_grad.py
+1
-1
tests/st/ops/ascend/test_dense_grad.py
tests/st/ops/ascend/test_dense_grad.py
+1
-1
tests/st/ops/ascend/test_fused_batchnorm_grad.py
tests/st/ops/ascend/test_fused_batchnorm_grad.py
+1
-1
tests/st/ops/ascend/test_maxpool_grad.py
tests/st/ops/ascend/test_maxpool_grad.py
+1
-1
tests/st/ops/ascend/test_maxpool_with_argmax_grad.py
tests/st/ops/ascend/test_maxpool_with_argmax_grad.py
+1
-1
tests/st/ops/ascend/test_relu_grad.py
tests/st/ops/ascend/test_relu_grad.py
+1
-1
tests/st/ops/ascend/test_simplemean_grad.py
tests/st/ops/ascend/test_simplemean_grad.py
+1
-1
tests/st/ops/ascend/test_tbe_ops/test_batchnorm_grad.py
tests/st/ops/ascend/test_tbe_ops/test_batchnorm_grad.py
+1
-1
tests/st/ops/ascend/test_tbe_ops/test_gelu_grad_sens.py
tests/st/ops/ascend/test_tbe_ops/test_gelu_grad_sens.py
+2
-2
tests/st/ops/ascend/test_tbe_ops/test_layernorm_grad.py
tests/st/ops/ascend/test_tbe_ops/test_layernorm_grad.py
+1
-1
tests/st/ops/ascend/test_tbe_ops/test_maximum_grad.py
tests/st/ops/ascend/test_tbe_ops/test_maximum_grad.py
+1
-1
tests/st/ops/ascend/test_tbe_ops/test_maxpool_grad.py
tests/st/ops/ascend/test_tbe_ops/test_maxpool_grad.py
+1
-1
tests/st/ops/ascend/test_tbe_ops/test_minimum_grad.py
tests/st/ops/ascend/test_tbe_ops/test_minimum_grad.py
+1
-1
tests/st/ops/ascend/test_tbe_ops/test_relu_grad.py
tests/st/ops/ascend/test_tbe_ops/test_relu_grad.py
+1
-1
tests/st/ops/ascend/test_tbe_ops/test_relu_v2_grad.py
tests/st/ops/ascend/test_tbe_ops/test_relu_v2_grad.py
+1
-1
tests/st/ops/ascend/test_tbe_ops/test_resize_nearest_neighbor_grad.py
.../ascend/test_tbe_ops/test_resize_nearest_neighbor_grad.py
+1
-1
tests/st/ops/ascend/test_tbe_ops/test_sigmoid_cross_entropy_with_logits_grad.py
...st_tbe_ops/test_sigmoid_cross_entropy_with_logits_grad.py
+1
-1
tests/st/ops/ascend/test_tbe_ops/test_sigmoid_grad.py
tests/st/ops/ascend/test_tbe_ops/test_sigmoid_grad.py
+1
-1
tests/st/ops/ascend/test_tbe_ops/test_smooth_l1_loss_grad.py
tests/st/ops/ascend/test_tbe_ops/test_smooth_l1_loss_grad.py
+1
-1
tests/st/ops/ascend/test_tbe_ops/test_stridedslice_grad.py
tests/st/ops/ascend/test_tbe_ops/test_stridedslice_grad.py
+1
-1
tests/st/ops/cpu/test_batchnorm_op.py
tests/st/ops/cpu/test_batchnorm_op.py
+1
-1
tests/st/ops/cpu/test_lstm_op.py
tests/st/ops/cpu/test_lstm_op.py
+1
-2
tests/st/ops/custom_ops_tbe/test_square.py
tests/st/ops/custom_ops_tbe/test_square.py
+1
-1
tests/st/ops/gpu/test_batchnorm_op.py
tests/st/ops/gpu/test_batchnorm_op.py
+1
-1
tests/st/ops/gpu/test_binary_cross_entropy_op.py
tests/st/ops/gpu/test_binary_cross_entropy_op.py
+1
-1
tests/st/ops/gpu/test_ctcloss_op.py
tests/st/ops/gpu/test_ctcloss_op.py
+1
-1
tests/st/ops/gpu/test_dense_op.py
tests/st/ops/gpu/test_dense_op.py
+3
-4
tests/st/ops/gpu/test_gelu_grad_op.py
tests/st/ops/gpu/test_gelu_grad_op.py
+1
-1
tests/st/ops/gpu/test_kl_div_op.py
tests/st/ops/gpu/test_kl_div_op.py
+1
-1
tests/st/ops/gpu/test_logsoftmax_op.py
tests/st/ops/gpu/test_logsoftmax_op.py
+1
-1
tests/st/ops/gpu/test_lstm_op.py
tests/st/ops/gpu/test_lstm_op.py
+1
-2
tests/st/ops/gpu/test_maximum_op.py
tests/st/ops/gpu/test_maximum_op.py
+1
-1
tests/st/ops/gpu/test_minimum_op.py
tests/st/ops/gpu/test_minimum_op.py
+1
-1
tests/st/ops/gpu/test_mirror_pad.py
tests/st/ops/gpu/test_mirror_pad.py
+1
-1
tests/st/ops/gpu/test_smoothl1loss_op.py
tests/st/ops/gpu/test_smoothl1loss_op.py
+1
-1
tests/st/ops/gpu/test_softmax_op.py
tests/st/ops/gpu/test_softmax_op.py
+1
-1
tests/st/ops/gpu/test_stridedslice_grad_op.py
tests/st/ops/gpu/test_stridedslice_grad_op.py
+1
-1
tests/st/ops/gpu/test_tanh_op.py
tests/st/ops/gpu/test_tanh_op.py
+1
-1
tests/st/pynative/test_pynative_hook.py
tests/st/pynative/test_pynative_hook.py
+2
-2
tests/st/pynative/test_pynative_lenet.py
tests/st/pynative/test_pynative_lenet.py
+1
-1
tests/st/pynative/test_pynative_mindarmour.py
tests/st/pynative/test_pynative_mindarmour.py
+2
-3
tests/st/pynative/test_pynative_resnet50.py
tests/st/pynative/test_pynative_resnet50.py
+1
-1
tests/st/pynative/test_tensor_index.py
tests/st/pynative/test_tensor_index.py
+1
-1
tests/train_step_wrap.py
tests/train_step_wrap.py
+3
-3
tests/ut/python/dtype/test_list.py
tests/ut/python/dtype/test_list.py
+1
-1
tests/ut/python/exec/test_train_with_lars.py
tests/ut/python/exec/test_train_with_lars.py
+1
-2
tests/ut/python/ir/test_row_tensor.py
tests/ut/python/ir/test_row_tensor.py
+2
-2
tests/ut/python/ir/test_sparse_tensor.py
tests/ut/python/ir/test_sparse_tensor.py
+1
-1
tests/ut/python/keep_order/test_keep_order.py
tests/ut/python/keep_order/test_keep_order.py
+1
-1
tests/ut/python/model/test_mix_precision.py
tests/ut/python/model/test_mix_precision.py
+1
-2
tests/ut/python/nn/optim/test_lr_schedule.py
tests/ut/python/nn/optim/test_lr_schedule.py
+1
-1
tests/ut/python/nn/test_nn_pad.py
tests/ut/python/nn/test_nn_pad.py
+1
-1
tests/ut/python/ops/test_bprop_disorder.py
tests/ut/python/ops/test_bprop_disorder.py
+1
-1
tests/ut/python/ops/test_control_ops.py
tests/ut/python/ops/test_control_ops.py
+5
-5
tests/ut/python/ops/test_math_ops.py
tests/ut/python/ops/test_math_ops.py
+1
-1
tests/ut/python/ops/test_momentum.py
tests/ut/python/ops/test_momentum.py
+1
-1
tests/ut/python/ops/test_nn_ops.py
tests/ut/python/ops/test_nn_ops.py
+3
-3
tests/ut/python/ops/test_ops.py
tests/ut/python/ops/test_ops.py
+1
-1
tests/ut/python/parallel/test_add_relu_redistribution.py
tests/ut/python/parallel/test_add_relu_redistribution.py
+1
-1
tests/ut/python/parallel/test_arithmetic.py
tests/ut/python/parallel/test_arithmetic.py
+1
-1
tests/ut/python/parallel/test_attention.py
tests/ut/python/parallel/test_attention.py
+1
-1
tests/ut/python/parallel/test_auto_parallel_BN_PReLU.py
tests/ut/python/parallel/test_auto_parallel_BN_PReLU.py
+1
-1
tests/ut/python/parallel/test_auto_parallel_arithmetic.py
tests/ut/python/parallel/test_auto_parallel_arithmetic.py
+1
-1
tests/ut/python/parallel/test_auto_parallel_assign_sub_with_ref_key.py
...on/parallel/test_auto_parallel_assign_sub_with_ref_key.py
+1
-1
tests/ut/python/parallel/test_auto_parallel_cast.py
tests/ut/python/parallel/test_auto_parallel_cast.py
+1
-1
tests/ut/python/parallel/test_auto_parallel_common_parameter.py
...ut/python/parallel/test_auto_parallel_common_parameter.py
+1
-1
tests/ut/python/parallel/test_auto_parallel_double_sources.py
...s/ut/python/parallel/test_auto_parallel_double_sources.py
+1
-1
tests/ut/python/parallel/test_auto_parallel_double_star.py
tests/ut/python/parallel/test_auto_parallel_double_star.py
+1
-1
tests/ut/python/parallel/test_auto_parallel_double_subgraphs.py
...ut/python/parallel/test_auto_parallel_double_subgraphs.py
+2
-2
tests/ut/python/parallel/test_auto_parallel_fc_nobias.py
tests/ut/python/parallel/test_auto_parallel_fc_nobias.py
+1
-1
tests/ut/python/parallel/test_auto_parallel_four_matmul.py
tests/ut/python/parallel/test_auto_parallel_four_matmul.py
+1
-1
tests/ut/python/parallel/test_auto_parallel_l2normalize.py
tests/ut/python/parallel/test_auto_parallel_l2normalize.py
+1
-1
tests/ut/python/parallel/test_auto_parallel_matmul_drop.py
tests/ut/python/parallel/test_auto_parallel_matmul_drop.py
+1
-1
tests/ut/python/parallel/test_auto_parallel_matmul_prelu.py
tests/ut/python/parallel/test_auto_parallel_matmul_prelu.py
+1
-1
tests/ut/python/parallel/test_auto_parallel_onehot.py
tests/ut/python/parallel/test_auto_parallel_onehot.py
+1
-1
tests/ut/python/parallel/test_auto_parallel_partial_strategy.py
...ut/python/parallel/test_auto_parallel_partial_strategy.py
+1
-1
tests/ut/python/parallel/test_auto_parallel_reduce_method.py
tests/ut/python/parallel/test_auto_parallel_reduce_method.py
+1
-1
tests/ut/python/parallel/test_auto_parallel_reshape.py
tests/ut/python/parallel/test_auto_parallel_reshape.py
+1
-1
tests/ut/python/parallel/test_auto_parallel_rhombus.py
tests/ut/python/parallel/test_auto_parallel_rhombus.py
+1
-1
tests/ut/python/parallel/test_auto_parallel_softmax_loss.py
tests/ut/python/parallel/test_auto_parallel_softmax_loss.py
+1
-1
tests/ut/python/parallel/test_auto_parallel_transformer.py
tests/ut/python/parallel/test_auto_parallel_transformer.py
+1
-1
tests/ut/python/parallel/test_auto_parallel_transpose.py
tests/ut/python/parallel/test_auto_parallel_transpose.py
+1
-1
tests/ut/python/parallel/test_auto_parallel_tuple_depend.py
tests/ut/python/parallel/test_auto_parallel_tuple_depend.py
+1
-1
tests/ut/python/parallel/test_auto_parallel_two_matmul.py
tests/ut/python/parallel/test_auto_parallel_two_matmul.py
+1
-1
tests/ut/python/parallel/test_auto_parallel_two_partial_matmul.py
.../python/parallel/test_auto_parallel_two_partial_matmul.py
+1
-1
tests/ut/python/parallel/test_auto_parallel_zig_zag.py
tests/ut/python/parallel/test_auto_parallel_zig_zag.py
+1
-1
tests/ut/python/parallel/test_auto_star_elimination.py
tests/ut/python/parallel/test_auto_star_elimination.py
+1
-1
tests/ut/python/parallel/test_batch_parallel.py
tests/ut/python/parallel/test_batch_parallel.py
+1
-1
tests/ut/python/parallel/test_batch_parallel_dropout.py
tests/ut/python/parallel/test_batch_parallel_dropout.py
+1
-1
tests/ut/python/parallel/test_batch_parallel_tensoradd.py
tests/ut/python/parallel/test_batch_parallel_tensoradd.py
+1
-1
tests/ut/python/parallel/test_comparison_function_info.py
tests/ut/python/parallel/test_comparison_function_info.py
+1
-1
tests/ut/python/parallel/test_dataset_interface.py
tests/ut/python/parallel/test_dataset_interface.py
+1
-1
tests/ut/python/parallel/test_different_type_for_div_op.py
tests/ut/python/parallel/test_different_type_for_div_op.py
+1
-1
tests/ut/python/parallel/test_element_wise_function.py
tests/ut/python/parallel/test_element_wise_function.py
+1
-1
tests/ut/python/parallel/test_embeddinglookup.py
tests/ut/python/parallel/test_embeddinglookup.py
+1
-1
tests/ut/python/parallel/test_gather_v2.py
tests/ut/python/parallel/test_gather_v2.py
+1
-1
tests/ut/python/parallel/test_gather_v2_primitive.py
tests/ut/python/parallel/test_gather_v2_primitive.py
+1
-2
tests/ut/python/parallel/test_get_next.py
tests/ut/python/parallel/test_get_next.py
+1
-1
tests/ut/python/parallel/test_gpu_dropout.py
tests/ut/python/parallel/test_gpu_dropout.py
+1
-1
tests/ut/python/parallel/test_hybird_parallel_activation.py
tests/ut/python/parallel/test_hybird_parallel_activation.py
+1
-1
tests/ut/python/parallel/test_l2normalize.py
tests/ut/python/parallel/test_l2normalize.py
+1
-1
tests/ut/python/parallel/test_linear.py
tests/ut/python/parallel/test_linear.py
+1
-1
tests/ut/python/parallel/test_loop_two_matmul.py
tests/ut/python/parallel/test_loop_two_matmul.py
+1
-1
tests/ut/python/parallel/test_loss_scale.py
tests/ut/python/parallel/test_loss_scale.py
+1
-2
tests/ut/python/parallel/test_matmul_dropout.py
tests/ut/python/parallel/test_matmul_dropout.py
+1
-1
tests/ut/python/parallel/test_matmul_tensor.py
tests/ut/python/parallel/test_matmul_tensor.py
+1
-1
tests/ut/python/parallel/test_mix_precision_hybrid_parallel.py
.../ut/python/parallel/test_mix_precision_hybrid_parallel.py
+1
-1
tests/ut/python/parallel/test_one_hot_net.py
tests/ut/python/parallel/test_one_hot_net.py
+1
-1
tests/ut/python/parallel/test_one_weight_parameter.py
tests/ut/python/parallel/test_one_weight_parameter.py
+1
-1
tests/ut/python/parallel/test_onehot.py
tests/ut/python/parallel/test_onehot.py
+1
-1
tests/ut/python/parallel/test_prelu.py
tests/ut/python/parallel/test_prelu.py
+1
-1
tests/ut/python/parallel/test_reduce_method_info.py
tests/ut/python/parallel/test_reduce_method_info.py
+1
-1
tests/ut/python/parallel/test_reshape.py
tests/ut/python/parallel/test_reshape.py
+2
-3
tests/ut/python/parallel/test_reshape_parameter.py
tests/ut/python/parallel/test_reshape_parameter.py
+1
-1
tests/ut/python/parallel/test_scalar_loss.py
tests/ut/python/parallel/test_scalar_loss.py
+1
-1
tests/ut/python/parallel/test_semi_auto_two_subgraphs.py
tests/ut/python/parallel/test_semi_auto_two_subgraphs.py
+2
-2
tests/ut/python/parallel/test_softmax_cross_entropy_loss.py
tests/ut/python/parallel/test_softmax_cross_entropy_loss.py
+1
-1
tests/ut/python/parallel/test_sparse_feature_bprop.py
tests/ut/python/parallel/test_sparse_feature_bprop.py
+1
-1
tests/ut/python/parallel/test_sparse_gather_v2.py
tests/ut/python/parallel/test_sparse_gather_v2.py
+1
-1
tests/ut/python/parallel/test_split_grad_sens.py
tests/ut/python/parallel/test_split_grad_sens.py
+2
-2
tests/ut/python/parallel/test_step_parallel.py
tests/ut/python/parallel/test_step_parallel.py
+1
-1
tests/ut/python/parallel/test_strategy_checkpoint.py
tests/ut/python/parallel/test_strategy_checkpoint.py
+1
-1
tests/ut/python/parallel/test_sum_as_loss.py
tests/ut/python/parallel/test_sum_as_loss.py
+1
-1
tests/ut/python/parallel/test_two_matmul.py
tests/ut/python/parallel/test_two_matmul.py
+1
-1
tests/ut/python/parallel/test_two_weights_parameter.py
tests/ut/python/parallel/test_two_weights_parameter.py
+1
-1
tests/ut/python/parallel/test_virtual_dataset_3_input.py
tests/ut/python/parallel/test_virtual_dataset_3_input.py
+1
-1
tests/ut/python/parameter_feature/test_parameter.py
tests/ut/python/parameter_feature/test_parameter.py
+2
-2
tests/ut/python/parameter_feature/test_var_grad.py
tests/ut/python/parameter_feature/test_var_grad.py
+8
-10
tests/ut/python/pipeline/infer/test_scalar_add_grad.py
tests/ut/python/pipeline/infer/test_scalar_add_grad.py
+1
-1
tests/ut/python/pipeline/parse/test_parse.py
tests/ut/python/pipeline/parse/test_parse.py
+1
-1
tests/ut/python/pynative_mode/ops/test_grad.py
tests/ut/python/pynative_mode/ops/test_grad.py
+2
-2
tests/ut/python/pynative_mode/test_cont_cases.py
tests/ut/python/pynative_mode/test_cont_cases.py
+2
-2
tests/ut/python/pynative_mode/test_framstruct.py
tests/ut/python/pynative_mode/test_framstruct.py
+4
-4
tests/ut/python/pynative_mode/test_high_order_grad.py
tests/ut/python/pynative_mode/test_high_order_grad.py
+3
-3
tests/ut/python/pynative_mode/test_hook.py
tests/ut/python/pynative_mode/test_hook.py
+2
-2
tests/ut/python/pynative_mode/test_implicit_conversion.py
tests/ut/python/pynative_mode/test_implicit_conversion.py
+1
-1
tests/ut/python/pynative_mode/test_insert_grad_of.py
tests/ut/python/pynative_mode/test_insert_grad_of.py
+2
-2
tests/ut/python/pynative_mode/test_kw_and_kwarg.py
tests/ut/python/pynative_mode/test_kw_and_kwarg.py
+2
-2
tests/ut/python/pynative_mode/test_pynative_model.py
tests/ut/python/pynative_mode/test_pynative_model.py
+1
-1
tests/ut/python/pynative_mode/test_sparse_pynative.py
tests/ut/python/pynative_mode/test_sparse_pynative.py
+1
-1
tests/ut/python/pynative_mode/test_stop_gradient.py
tests/ut/python/pynative_mode/test_stop_gradient.py
+2
-2
tests/ut/python/pynative_mode/test_user_define_bprop_check.py
...s/ut/python/pynative_mode/test_user_define_bprop_check.py
+1
-1
未找到文件。
mindspore/nn/wrap/cell_wrapper.py
浏览文件 @
1a54785f
...
@@ -117,7 +117,7 @@ class WithGradCell(Cell):
...
@@ -117,7 +117,7 @@ class WithGradCell(Cell):
self
.
network
=
network
self
.
network
=
network
self
.
loss_fn
=
loss_fn
self
.
loss_fn
=
loss_fn
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
grad
=
C
.
GradOperation
(
'grad'
,
get_by_list
=
True
,
sens_param
=
(
sens
is
not
None
))
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
(
sens
is
not
None
))
self
.
sens
=
sens
self
.
sens
=
sens
if
loss_fn
is
None
:
if
loss_fn
is
None
:
self
.
network_with_loss
=
network
self
.
network_with_loss
=
network
...
@@ -182,7 +182,7 @@ class TrainOneStepCell(Cell):
...
@@ -182,7 +182,7 @@ class TrainOneStepCell(Cell):
self
.
network
.
add_flags
(
defer_inline
=
True
)
self
.
network
.
add_flags
(
defer_inline
=
True
)
self
.
weights
=
optimizer
.
parameters
self
.
weights
=
optimizer
.
parameters
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
get_by_list
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
self
.
sens
=
sens
self
.
sens
=
sens
self
.
reducer_flag
=
False
self
.
reducer_flag
=
False
self
.
grad_reducer
=
None
self
.
grad_reducer
=
None
...
...
mindspore/nn/wrap/grad_reducer.py
浏览文件 @
1a54785f
...
@@ -269,7 +269,7 @@ class DistributedGradReducer(Cell):
...
@@ -269,7 +269,7 @@ class DistributedGradReducer(Cell):
>>> self.network.add_flags(defer_inline=True)
>>> self.network.add_flags(defer_inline=True)
>>> self.weights = optimizer.parameters
>>> self.weights = optimizer.parameters
>>> self.optimizer = optimizer
>>> self.optimizer = optimizer
>>> self.grad = C.GradOperation(
'grad',
get_by_list=True, sens_param=True)
>>> self.grad = C.GradOperation(get_by_list=True, sens_param=True)
>>> self.sens = sens
>>> self.sens = sens
>>> self.reducer_flag = False
>>> self.reducer_flag = False
>>> self.grad_reducer = None
>>> self.grad_reducer = None
...
...
mindspore/nn/wrap/loss_scale.py
浏览文件 @
1a54785f
...
@@ -210,7 +210,7 @@ class TrainOneStepWithLossScaleCell(Cell):
...
@@ -210,7 +210,7 @@ class TrainOneStepWithLossScaleCell(Cell):
self
.
network
.
add_flags
(
defer_inline
=
True
)
self
.
network
.
add_flags
(
defer_inline
=
True
)
self
.
weights
=
optimizer
.
parameters
self
.
weights
=
optimizer
.
parameters
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
get_by_list
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
self
.
hyper_map
=
C
.
HyperMap
()
self
.
hyper_map
=
C
.
HyperMap
()
if
context
.
get_context
(
"device_target"
)
==
"GPU"
:
if
context
.
get_context
(
"device_target"
)
==
"GPU"
:
self
.
gpu_target
=
True
self
.
gpu_target
=
True
...
...
mindspore/ops/composite/base.py
浏览文件 @
1a54785f
...
@@ -106,12 +106,11 @@ class GradOperation(GradOperation_):
...
@@ -106,12 +106,11 @@ class GradOperation(GradOperation_):
a 'ones_like(outputs)' sensitivity will be attached automatically. Default: False.
a 'ones_like(outputs)' sensitivity will be attached automatically. Default: False.
"""
"""
def
__init__
(
self
,
name
,
def
__init__
(
self
,
get_all
=
False
,
get_by_list
=
False
,
sens_param
=
False
):
get_all
=
False
,
get_by_list
=
False
,
sens_param
=
False
):
self
.
get_all
=
get_all
self
.
get_all
=
get_all
self
.
get_by_list
=
get_by_list
self
.
get_by_list
=
get_by_list
self
.
sens_param
=
sens_param
self
.
sens_param
=
sens_param
GradOperation_
.
__init__
(
self
,
name
,
get_all
,
get_by_list
,
sens_param
)
GradOperation_
.
__init__
(
self
,
'grad'
,
get_all
,
get_by_list
,
sens_param
)
self
.
grad_fn
=
None
self
.
grad_fn
=
None
self
.
fn
=
None
self
.
fn
=
None
self
.
need_forward
=
False
self
.
need_forward
=
False
...
@@ -139,7 +138,7 @@ class GradOperation(GradOperation_):
...
@@ -139,7 +138,7 @@ class GradOperation(GradOperation_):
fn
.
already_run
=
False
fn
.
already_run
=
False
def
__call__
(
self
,
fn
,
weights
=
None
):
def
__call__
(
self
,
fn
,
weights
=
None
):
grad_
=
GradOperation
(
'grad'
,
self
.
get_all
,
self
.
get_by_list
,
self
.
sens_param
)
grad_
=
GradOperation
(
self
.
get_all
,
self
.
get_by_list
,
self
.
sens_param
)
if
self
.
grad_fn
is
None
or
self
.
fn
!=
fn
:
if
self
.
grad_fn
is
None
or
self
.
fn
!=
fn
:
if
context
.
get_context
(
"mode"
)
==
context
.
GRAPH_MODE
:
if
context
.
get_context
(
"mode"
)
==
context
.
GRAPH_MODE
:
if
self
.
get_by_list
:
if
self
.
get_by_list
:
...
...
mindspore/ops/operations/debug_ops.py
浏览文件 @
1a54785f
...
@@ -216,7 +216,7 @@ class InsertGradientOf(PrimitiveWithInfer):
...
@@ -216,7 +216,7 @@ class InsertGradientOf(PrimitiveWithInfer):
>>> return ret
>>> return ret
>>>
>>>
>>> clip = P.InsertGradientOf(clip_gradient)
>>> clip = P.InsertGradientOf(clip_gradient)
>>> grad_all = C.GradOperation(
'get_all',
get_all=True)
>>> grad_all = C.GradOperation(get_all=True)
>>> def InsertGradientOfClipDemo():
>>> def InsertGradientOfClipDemo():
>>> def clip_test(x, y):
>>> def clip_test(x, y):
>>> x = clip(x)
>>> x = clip(x)
...
@@ -268,7 +268,7 @@ class HookBackward(PrimitiveWithInfer):
...
@@ -268,7 +268,7 @@ class HookBackward(PrimitiveWithInfer):
>>> def hook_fn(grad_out):
>>> def hook_fn(grad_out):
>>> print(grad_out)
>>> print(grad_out)
>>>
>>>
>>> grad_all = GradOperation(
'get_all',
get_all=True)
>>> grad_all = GradOperation(get_all=True)
>>> hook = P.HookBackward(hook_fn)
>>> hook = P.HookBackward(hook_fn)
>>>
>>>
>>> def hook_test(x, y):
>>> def hook_test(x, y):
...
...
model_zoo/official/cv/faster_rcnn/src/network_define.py
浏览文件 @
1a54785f
...
@@ -163,8 +163,7 @@ class TrainOneStepCell(nn.Cell):
...
@@ -163,8 +163,7 @@ class TrainOneStepCell(nn.Cell):
self
.
backbone
=
network_backbone
self
.
backbone
=
network_backbone
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
get_by_list
=
True
,
sens_param
=
True
)
sens_param
=
True
)
self
.
sens
=
Tensor
((
np
.
ones
((
1
,))
*
sens
).
astype
(
np
.
float16
))
self
.
sens
=
Tensor
((
np
.
ones
((
1
,))
*
sens
).
astype
(
np
.
float16
))
self
.
reduce_flag
=
reduce_flag
self
.
reduce_flag
=
reduce_flag
...
...
model_zoo/official/cv/maskrcnn/src/network_define.py
浏览文件 @
1a54785f
...
@@ -171,8 +171,7 @@ class TrainOneStepCell(nn.Cell):
...
@@ -171,8 +171,7 @@ class TrainOneStepCell(nn.Cell):
self
.
backbone
=
network_backbone
self
.
backbone
=
network_backbone
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
get_by_list
=
True
,
sens_param
=
True
)
sens_param
=
True
)
self
.
sens
=
Tensor
((
np
.
ones
((
1
,))
*
sens
).
astype
(
np
.
float16
))
self
.
sens
=
Tensor
((
np
.
ones
((
1
,))
*
sens
).
astype
(
np
.
float16
))
self
.
reduce_flag
=
reduce_flag
self
.
reduce_flag
=
reduce_flag
...
...
model_zoo/official/cv/resnet_thor/src/grad_reducer_thor.py
浏览文件 @
1a54785f
...
@@ -119,7 +119,7 @@ class DistributedGradReducerThor(Cell):
...
@@ -119,7 +119,7 @@ class DistributedGradReducerThor(Cell):
>>> self.network.add_flags(defer_inline=True)
>>> self.network.add_flags(defer_inline=True)
>>> self.weights = ParameterTuple(network.trainable_params())
>>> self.weights = ParameterTuple(network.trainable_params())
>>> self.optimizer = optimizer
>>> self.optimizer = optimizer
>>> self.grad = C.GradOperation(
'grad',
get_by_list=True, sens_param=True)
>>> self.grad = C.GradOperation(get_by_list=True, sens_param=True)
>>> self.sens = sens
>>> self.sens = sens
>>> self.reducer_flag = False
>>> self.reducer_flag = False
>>> self.grad_reducer = None
>>> self.grad_reducer = None
...
...
model_zoo/official/cv/ssd/src/ssd.py
浏览文件 @
1a54785f
...
@@ -383,7 +383,7 @@ class TrainingWrapper(nn.Cell):
...
@@ -383,7 +383,7 @@ class TrainingWrapper(nn.Cell):
self
.
network
=
network
self
.
network
=
network
self
.
weights
=
ms
.
ParameterTuple
(
network
.
trainable_params
())
self
.
weights
=
ms
.
ParameterTuple
(
network
.
trainable_params
())
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
get_by_list
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
self
.
sens
=
sens
self
.
sens
=
sens
self
.
reducer_flag
=
False
self
.
reducer_flag
=
False
self
.
grad_reducer
=
None
self
.
grad_reducer
=
None
...
...
model_zoo/official/cv/warpctc/src/warpctc_for_train.py
浏览文件 @
1a54785f
...
@@ -77,7 +77,7 @@ class TrainOneStepCellWithGradClip(Cell):
...
@@ -77,7 +77,7 @@ class TrainOneStepCellWithGradClip(Cell):
self
.
network
.
add_flags
(
defer_inline
=
True
)
self
.
network
.
add_flags
(
defer_inline
=
True
)
self
.
weights
=
optimizer
.
parameters
self
.
weights
=
optimizer
.
parameters
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
get_by_list
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
self
.
sens
=
sens
self
.
sens
=
sens
self
.
reducer_flag
=
False
self
.
reducer_flag
=
False
self
.
grad_reducer
=
None
self
.
grad_reducer
=
None
...
...
model_zoo/official/cv/yolov3_darknet53/src/yolo.py
浏览文件 @
1a54785f
...
@@ -412,7 +412,7 @@ class TrainingWrapper(nn.Cell):
...
@@ -412,7 +412,7 @@ class TrainingWrapper(nn.Cell):
self
.
network
=
network
self
.
network
=
network
self
.
weights
=
optimizer
.
parameters
self
.
weights
=
optimizer
.
parameters
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
get_by_list
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
self
.
sens
=
sens
self
.
sens
=
sens
self
.
reducer_flag
=
False
self
.
reducer_flag
=
False
self
.
grad_reducer
=
None
self
.
grad_reducer
=
None
...
...
model_zoo/official/cv/yolov3_darknet53_quant/src/yolo.py
浏览文件 @
1a54785f
...
@@ -412,7 +412,7 @@ class TrainingWrapper(nn.Cell):
...
@@ -412,7 +412,7 @@ class TrainingWrapper(nn.Cell):
self
.
network
=
network
self
.
network
=
network
self
.
weights
=
optimizer
.
parameters
self
.
weights
=
optimizer
.
parameters
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
get_by_list
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
self
.
sens
=
sens
self
.
sens
=
sens
self
.
reducer_flag
=
False
self
.
reducer_flag
=
False
self
.
grad_reducer
=
None
self
.
grad_reducer
=
None
...
...
model_zoo/official/cv/yolov3_resnet18/src/yolov3.py
浏览文件 @
1a54785f
...
@@ -647,7 +647,7 @@ class TrainingWrapper(nn.Cell):
...
@@ -647,7 +647,7 @@ class TrainingWrapper(nn.Cell):
self
.
network
=
network
self
.
network
=
network
self
.
weights
=
ms
.
ParameterTuple
(
network
.
trainable_params
())
self
.
weights
=
ms
.
ParameterTuple
(
network
.
trainable_params
())
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
get_by_list
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
self
.
sens
=
sens
self
.
sens
=
sens
self
.
reducer_flag
=
False
self
.
reducer_flag
=
False
self
.
grad_reducer
=
None
self
.
grad_reducer
=
None
...
...
model_zoo/official/gnn/gat/src/utils.py
浏览文件 @
1a54785f
...
@@ -141,7 +141,7 @@ class TrainOneStepCell(nn.Cell):
...
@@ -141,7 +141,7 @@ class TrainOneStepCell(nn.Cell):
self
.
network
.
add_flags
(
defer_inline
=
True
)
self
.
network
.
add_flags
(
defer_inline
=
True
)
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
get_by_list
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
self
.
sens
=
sens
self
.
sens
=
sens
def
construct
(
self
):
def
construct
(
self
):
...
...
model_zoo/official/gnn/gcn/src/metrics.py
浏览文件 @
1a54785f
...
@@ -150,7 +150,7 @@ class TrainOneStepCell(nn.Cell):
...
@@ -150,7 +150,7 @@ class TrainOneStepCell(nn.Cell):
self
.
network
.
add_flags
(
defer_inline
=
True
)
self
.
network
.
add_flags
(
defer_inline
=
True
)
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
get_by_list
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
self
.
sens
=
sens
self
.
sens
=
sens
def
construct
(
self
):
def
construct
(
self
):
...
...
model_zoo/official/nlp/bert/src/bert_for_finetune.py
浏览文件 @
1a54785f
...
@@ -57,8 +57,7 @@ class BertFinetuneCell(nn.Cell):
...
@@ -57,8 +57,7 @@ class BertFinetuneCell(nn.Cell):
self
.
network
=
network
self
.
network
=
network
self
.
weights
=
optimizer
.
parameters
self
.
weights
=
optimizer
.
parameters
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
get_by_list
=
True
,
sens_param
=
True
)
sens_param
=
True
)
self
.
reducer_flag
=
False
self
.
reducer_flag
=
False
self
.
allreduce
=
P
.
AllReduce
()
self
.
allreduce
=
P
.
AllReduce
()
...
@@ -160,7 +159,7 @@ class BertSquadCell(nn.Cell):
...
@@ -160,7 +159,7 @@ class BertSquadCell(nn.Cell):
self
.
network
=
network
self
.
network
=
network
self
.
weights
=
optimizer
.
parameters
self
.
weights
=
optimizer
.
parameters
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
get_by_list
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
self
.
reducer_flag
=
False
self
.
reducer_flag
=
False
self
.
allreduce
=
P
.
AllReduce
()
self
.
allreduce
=
P
.
AllReduce
()
self
.
parallel_mode
=
context
.
get_auto_parallel_context
(
"parallel_mode"
)
self
.
parallel_mode
=
context
.
get_auto_parallel_context
(
"parallel_mode"
)
...
...
model_zoo/official/nlp/bert/src/bert_for_pre_training.py
浏览文件 @
1a54785f
...
@@ -274,7 +274,7 @@ class BertTrainOneStepCell(nn.Cell):
...
@@ -274,7 +274,7 @@ class BertTrainOneStepCell(nn.Cell):
self
.
network
=
network
self
.
network
=
network
self
.
weights
=
optimizer
.
parameters
self
.
weights
=
optimizer
.
parameters
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
get_by_list
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
self
.
sens
=
sens
self
.
sens
=
sens
self
.
reducer_flag
=
False
self
.
reducer_flag
=
False
self
.
parallel_mode
=
context
.
get_auto_parallel_context
(
"parallel_mode"
)
self
.
parallel_mode
=
context
.
get_auto_parallel_context
(
"parallel_mode"
)
...
@@ -353,8 +353,7 @@ class BertTrainOneStepWithLossScaleCell(nn.Cell):
...
@@ -353,8 +353,7 @@ class BertTrainOneStepWithLossScaleCell(nn.Cell):
self
.
network
=
network
self
.
network
=
network
self
.
weights
=
optimizer
.
parameters
self
.
weights
=
optimizer
.
parameters
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
get_by_list
=
True
,
sens_param
=
True
)
sens_param
=
True
)
self
.
reducer_flag
=
False
self
.
reducer_flag
=
False
self
.
allreduce
=
P
.
AllReduce
()
self
.
allreduce
=
P
.
AllReduce
()
...
...
model_zoo/official/nlp/bert_thor/src/bert_for_pre_training.py
浏览文件 @
1a54785f
...
@@ -293,7 +293,7 @@ class BertTrainOneStepCell(nn.Cell):
...
@@ -293,7 +293,7 @@ class BertTrainOneStepCell(nn.Cell):
self
.
network
=
network
self
.
network
=
network
self
.
weights
=
optimizer
.
parameters
self
.
weights
=
optimizer
.
parameters
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
get_by_list
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
self
.
sens
=
sens
self
.
sens
=
sens
self
.
reducer_flag
=
False
self
.
reducer_flag
=
False
self
.
parallel_mode
=
context
.
get_auto_parallel_context
(
"parallel_mode"
)
self
.
parallel_mode
=
context
.
get_auto_parallel_context
(
"parallel_mode"
)
...
@@ -373,8 +373,7 @@ class BertTrainOneStepWithLossScaleCell(nn.Cell):
...
@@ -373,8 +373,7 @@ class BertTrainOneStepWithLossScaleCell(nn.Cell):
self
.
network
=
network
self
.
network
=
network
self
.
weights
=
optimizer
.
parameters
self
.
weights
=
optimizer
.
parameters
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
get_by_list
=
True
,
sens_param
=
True
)
sens_param
=
True
)
self
.
reducer_flag
=
False
self
.
reducer_flag
=
False
self
.
allreduce
=
P
.
AllReduce
()
self
.
allreduce
=
P
.
AllReduce
()
...
...
model_zoo/official/nlp/bert_thor/src/grad_reducer_thor.py
浏览文件 @
1a54785f
...
@@ -119,7 +119,7 @@ class DistributedGradReducerThor(Cell):
...
@@ -119,7 +119,7 @@ class DistributedGradReducerThor(Cell):
>>> self.network.add_flags(defer_inline=True)
>>> self.network.add_flags(defer_inline=True)
>>> self.weights = ParameterTuple(network.trainable_params())
>>> self.weights = ParameterTuple(network.trainable_params())
>>> self.optimizer = optimizer
>>> self.optimizer = optimizer
>>> self.grad = C.GradOperation(
'grad',
get_by_list=True, sens_param=True)
>>> self.grad = C.GradOperation(get_by_list=True, sens_param=True)
>>> self.sens = sens
>>> self.sens = sens
>>> self.reducer_flag = False
>>> self.reducer_flag = False
>>> self.grad_reducer = None
>>> self.grad_reducer = None
...
...
model_zoo/official/nlp/mass/src/transformer/transformer_for_train.py
浏览文件 @
1a54785f
...
@@ -239,7 +239,7 @@ class TransformerTrainOneStepWithLossScaleCell(nn.Cell):
...
@@ -239,7 +239,7 @@ class TransformerTrainOneStepWithLossScaleCell(nn.Cell):
self
.
network
.
add_flags
(
defer_inline
=
True
)
self
.
network
.
add_flags
(
defer_inline
=
True
)
self
.
weights
=
optimizer
.
parameters
self
.
weights
=
optimizer
.
parameters
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
get_by_list
=
True
,
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
sens_param
=
True
)
self
.
reducer_flag
=
False
self
.
reducer_flag
=
False
self
.
all_reduce
=
P
.
AllReduce
()
self
.
all_reduce
=
P
.
AllReduce
()
...
...
model_zoo/official/nlp/tinybert/src/tinybert_for_gd_td.py
浏览文件 @
1a54785f
...
@@ -218,8 +218,7 @@ class BertTrainWithLossScaleCell(nn.Cell):
...
@@ -218,8 +218,7 @@ class BertTrainWithLossScaleCell(nn.Cell):
self
.
network
=
network
self
.
network
=
network
self
.
weights
=
optimizer
.
parameters
self
.
weights
=
optimizer
.
parameters
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
get_by_list
=
True
,
sens_param
=
True
)
sens_param
=
True
)
self
.
reducer_flag
=
False
self
.
reducer_flag
=
False
self
.
allreduce
=
P
.
AllReduce
()
self
.
allreduce
=
P
.
AllReduce
()
...
@@ -310,8 +309,7 @@ class BertTrainCell(nn.Cell):
...
@@ -310,8 +309,7 @@ class BertTrainCell(nn.Cell):
self
.
weights
=
optimizer
.
parameters
self
.
weights
=
optimizer
.
parameters
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
sens
=
sens
self
.
sens
=
sens
self
.
grad
=
C
.
GradOperation
(
'grad'
,
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
get_by_list
=
True
,
sens_param
=
True
)
sens_param
=
True
)
self
.
reducer_flag
=
False
self
.
reducer_flag
=
False
self
.
parallel_mode
=
context
.
get_auto_parallel_context
(
"parallel_mode"
)
self
.
parallel_mode
=
context
.
get_auto_parallel_context
(
"parallel_mode"
)
...
@@ -474,8 +472,7 @@ class BertEvaluationWithLossScaleCell(nn.Cell):
...
@@ -474,8 +472,7 @@ class BertEvaluationWithLossScaleCell(nn.Cell):
self
.
network
=
network
self
.
network
=
network
self
.
weights
=
optimizer
.
parameters
self
.
weights
=
optimizer
.
parameters
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
get_by_list
=
True
,
sens_param
=
True
)
sens_param
=
True
)
self
.
reducer_flag
=
False
self
.
reducer_flag
=
False
self
.
allreduce
=
P
.
AllReduce
()
self
.
allreduce
=
P
.
AllReduce
()
...
@@ -562,8 +559,7 @@ class BertEvaluationCell(nn.Cell):
...
@@ -562,8 +559,7 @@ class BertEvaluationCell(nn.Cell):
self
.
weights
=
optimizer
.
parameters
self
.
weights
=
optimizer
.
parameters
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
sens
=
sens
self
.
sens
=
sens
self
.
grad
=
C
.
GradOperation
(
'grad'
,
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
get_by_list
=
True
,
sens_param
=
True
)
sens_param
=
True
)
self
.
reducer_flag
=
False
self
.
reducer_flag
=
False
self
.
parallel_mode
=
context
.
get_auto_parallel_context
(
"parallel_mode"
)
self
.
parallel_mode
=
context
.
get_auto_parallel_context
(
"parallel_mode"
)
...
...
model_zoo/official/nlp/transformer/src/transformer_for_train.py
浏览文件 @
1a54785f
...
@@ -158,7 +158,7 @@ class TransformerTrainOneStepCell(nn.Cell):
...
@@ -158,7 +158,7 @@ class TransformerTrainOneStepCell(nn.Cell):
self
.
network
=
network
self
.
network
=
network
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
get_by_list
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
self
.
sens
=
sens
self
.
sens
=
sens
self
.
reducer_flag
=
False
self
.
reducer_flag
=
False
self
.
parallel_mode
=
context
.
get_auto_parallel_context
(
"parallel_mode"
)
self
.
parallel_mode
=
context
.
get_auto_parallel_context
(
"parallel_mode"
)
...
@@ -244,8 +244,7 @@ class TransformerTrainOneStepWithLossScaleCell(nn.Cell):
...
@@ -244,8 +244,7 @@ class TransformerTrainOneStepWithLossScaleCell(nn.Cell):
self
.
network
.
add_flags
(
defer_inline
=
True
)
self
.
network
.
add_flags
(
defer_inline
=
True
)
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
get_by_list
=
True
,
sens_param
=
True
)
sens_param
=
True
)
self
.
reducer_flag
=
False
self
.
reducer_flag
=
False
self
.
allreduce
=
P
.
AllReduce
()
self
.
allreduce
=
P
.
AllReduce
()
...
...
model_zoo/official/recommend/deepfm/src/deepfm.py
浏览文件 @
1a54785f
...
@@ -286,7 +286,7 @@ class TrainStepWrap(nn.Cell):
...
@@ -286,7 +286,7 @@ class TrainStepWrap(nn.Cell):
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
optimizer
=
Adam
(
self
.
weights
,
learning_rate
=
lr
,
eps
=
eps
,
loss_scale
=
loss_scale
)
self
.
optimizer
=
Adam
(
self
.
weights
,
learning_rate
=
lr
,
eps
=
eps
,
loss_scale
=
loss_scale
)
self
.
hyper_map
=
C
.
HyperMap
()
self
.
hyper_map
=
C
.
HyperMap
()
self
.
grad
=
C
.
GradOperation
(
'grad'
,
get_by_list
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
self
.
sens
=
loss_scale
self
.
sens
=
loss_scale
def
construct
(
self
,
batch_ids
,
batch_wts
,
label
):
def
construct
(
self
,
batch_ids
,
batch_wts
,
label
):
...
...
model_zoo/official/recommend/wide_and_deep/src/wide_and_deep.py
浏览文件 @
1a54785f
...
@@ -337,9 +337,9 @@ class TrainStepWrap(nn.Cell):
...
@@ -337,9 +337,9 @@ class TrainStepWrap(nn.Cell):
self
.
optimizer_w
=
FTRL
(
learning_rate
=
5e-2
,
params
=
self
.
weights_w
,
self
.
optimizer_w
=
FTRL
(
learning_rate
=
5e-2
,
params
=
self
.
weights_w
,
l1
=
1e-8
,
l2
=
1e-8
,
initial_accum
=
1.0
,
loss_scale
=
sens
)
l1
=
1e-8
,
l2
=
1e-8
,
initial_accum
=
1.0
,
loss_scale
=
sens
)
self
.
hyper_map
=
C
.
HyperMap
()
self
.
hyper_map
=
C
.
HyperMap
()
self
.
grad_w
=
C
.
GradOperation
(
'grad_w'
,
get_by_list
=
True
,
self
.
grad_w
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
sens_param
=
True
)
self
.
grad_d
=
C
.
GradOperation
(
'grad_d'
,
get_by_list
=
True
,
self
.
grad_d
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
sens_param
=
True
)
self
.
sens
=
sens
self
.
sens
=
sens
self
.
loss_net_w
=
IthOutputCell
(
network
,
output_index
=
0
)
self
.
loss_net_w
=
IthOutputCell
(
network
,
output_index
=
0
)
...
...
model_zoo/official/recommend/wide_and_deep_multitable/src/wide_and_deep.py
浏览文件 @
1a54785f
...
@@ -537,11 +537,9 @@ class TrainStepWrap(nn.Cell):
...
@@ -537,11 +537,9 @@ class TrainStepWrap(nn.Cell):
self
.
hyper_map
=
C
.
HyperMap
()
self
.
hyper_map
=
C
.
HyperMap
()
self
.
grad_w
=
C
.
GradOperation
(
'grad_w'
,
self
.
grad_w
=
C
.
GradOperation
(
get_by_list
=
True
,
get_by_list
=
True
,
sens_param
=
True
)
sens_param
=
True
)
self
.
grad_d
=
C
.
GradOperation
(
'grad_d'
,
self
.
grad_d
=
C
.
GradOperation
(
get_by_list
=
True
,
get_by_list
=
True
,
sens_param
=
True
)
sens_param
=
True
)
self
.
sens
=
sens
self
.
sens
=
sens
...
...
tests/mindspore_test_framework/components/function/compile_gradient_wrt_inputs.py
浏览文件 @
1a54785f
...
@@ -46,5 +46,5 @@ class CompileBackwardBlockWrtInputsBC(IBuilderComponent):
...
@@ -46,5 +46,5 @@ class CompileBackwardBlockWrtInputsBC(IBuilderComponent):
"""
"""
def
__call__
(
self
):
def
__call__
(
self
):
grad_op
=
GradOperation
(
'grad'
,
get_all
=
True
,
sens_param
=
True
)
grad_op
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
return
create_funcs
(
self
.
verification_set
,
gen_grad_net
,
compile_block
,
grad_op
)
return
create_funcs
(
self
.
verification_set
,
gen_grad_net
,
compile_block
,
grad_op
)
tests/mindspore_test_framework/components/function/compile_gradient_wrt_params.py
浏览文件 @
1a54785f
...
@@ -46,5 +46,5 @@ class CompileBackwardBlockWrtParamsBC(IBuilderComponent):
...
@@ -46,5 +46,5 @@ class CompileBackwardBlockWrtParamsBC(IBuilderComponent):
"""
"""
def
__call__
(
self
,
verification_set
):
def
__call__
(
self
,
verification_set
):
grad_op
=
GradOperation
(
'grad'
,
get_by_list
=
True
,
sens_param
=
True
)
grad_op
=
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
return
create_funcs
(
self
.
verification_set
,
gen_grad_net
,
compile_block
,
grad_op
)
return
create_funcs
(
self
.
verification_set
,
gen_grad_net
,
compile_block
,
grad_op
)
tests/mindspore_test_framework/components/function/init_params_with_rand_and_run_gradient_wrt_inputs.py
浏览文件 @
1a54785f
...
@@ -22,5 +22,5 @@ from ...utils.block_util import run_block, gen_grad_net, create_funcs, get_unifo
...
@@ -22,5 +22,5 @@ from ...utils.block_util import run_block, gen_grad_net, create_funcs, get_unifo
class
RunBackwardBlockWrtInputsWithRandParamBC
(
IBuilderComponent
):
class
RunBackwardBlockWrtInputsWithRandParamBC
(
IBuilderComponent
):
def
__call__
(
self
):
def
__call__
(
self
):
grad_op
=
GradOperation
(
'grad'
,
get_all
=
True
,
sens_param
=
True
)
grad_op
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
return
create_funcs
(
self
.
verification_set
,
gen_grad_net
,
run_block
,
grad_op
,
get_uniform_with_shape
)
return
create_funcs
(
self
.
verification_set
,
gen_grad_net
,
run_block
,
grad_op
,
get_uniform_with_shape
)
tests/mindspore_test_framework/components/function/init_params_with_rand_and_run_gradient_wrt_params.py
浏览文件 @
1a54785f
...
@@ -22,5 +22,5 @@ from ...utils.block_util import run_block, gen_grad_net, create_funcs, get_unifo
...
@@ -22,5 +22,5 @@ from ...utils.block_util import run_block, gen_grad_net, create_funcs, get_unifo
class
RunBackwardBlockWrtParamsWithRandParamBC
(
IBuilderComponent
):
class
RunBackwardBlockWrtParamsWithRandParamBC
(
IBuilderComponent
):
def
__call__
(
self
):
def
__call__
(
self
):
grad_op
=
GradOperation
(
'grad'
,
get_by_list
=
True
,
sens_param
=
True
)
grad_op
=
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
return
create_funcs
(
self
.
verification_set
,
gen_grad_net
,
run_block
,
grad_op
,
get_uniform_with_shape
)
return
create_funcs
(
self
.
verification_set
,
gen_grad_net
,
run_block
,
grad_op
,
get_uniform_with_shape
)
tests/mindspore_test_framework/components/function/run_gradient_wrt_inputs.py
浏览文件 @
1a54785f
...
@@ -22,5 +22,5 @@ from ...utils.block_util import run_block, gen_grad_net, create_funcs
...
@@ -22,5 +22,5 @@ from ...utils.block_util import run_block, gen_grad_net, create_funcs
class
RunBackwardBlockWrtInputsBC
(
IBuilderComponent
):
class
RunBackwardBlockWrtInputsBC
(
IBuilderComponent
):
def
__call__
(
self
):
def
__call__
(
self
):
grad_op
=
GradOperation
(
'grad'
,
get_all
=
True
,
sens_param
=
True
)
grad_op
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
return
create_funcs
(
self
.
verification_set
,
gen_grad_net
,
run_block
,
grad_op
)
return
create_funcs
(
self
.
verification_set
,
gen_grad_net
,
run_block
,
grad_op
)
tests/mindspore_test_framework/components/function/run_gradient_wrt_params.py
浏览文件 @
1a54785f
...
@@ -22,5 +22,5 @@ from ...utils.block_util import run_block, gen_grad_net, create_funcs
...
@@ -22,5 +22,5 @@ from ...utils.block_util import run_block, gen_grad_net, create_funcs
class
RunBackwardBlockWrtParamsBC
(
IBuilderComponent
):
class
RunBackwardBlockWrtParamsBC
(
IBuilderComponent
):
def
__call__
(
self
):
def
__call__
(
self
):
grad_op
=
GradOperation
(
'grad'
,
get_by_list
=
True
,
sens_param
=
True
)
grad_op
=
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
return
create_funcs
(
self
.
verification_set
,
gen_grad_net
,
run_block
,
grad_op
)
return
create_funcs
(
self
.
verification_set
,
gen_grad_net
,
run_block
,
grad_op
)
tests/mindspore_test_framework/utils/block_util.py
浏览文件 @
1a54785f
...
@@ -331,7 +331,7 @@ def create_funcs(verification_set, block_generator, block_runner, grad_op=None,
...
@@ -331,7 +331,7 @@ def create_funcs(verification_set, block_generator, block_runner, grad_op=None,
# gradient
# gradient
if
grad_op
:
if
grad_op
:
if
num_outputs
==
0
:
if
num_outputs
==
0
:
grad_op_
=
GradOperation
(
'grad'
,
get_all
=
grad_op
.
get_all
,
grad_op_
=
GradOperation
(
get_all
=
grad_op
.
get_all
,
get_by_list
=
grad_op
.
get_by_list
,
sens_param
=
False
)
get_by_list
=
grad_op
.
get_by_list
,
sens_param
=
False
)
b
=
block_generator
(
block
,
grad_op_
,
len
(
inputs
),
desc_const
=
desc_const
,
b
=
block_generator
(
block
,
grad_op_
,
len
(
inputs
),
desc_const
=
desc_const
,
const_first
=
const_first
,
add_fake_input
=
add_fake_input
)
const_first
=
const_first
,
add_fake_input
=
add_fake_input
)
...
...
tests/mindspore_test_framework/utils/bprop_util.py
浏览文件 @
1a54785f
...
@@ -85,7 +85,7 @@ def bprop(func, *inputs, grads_wrt_outputs=None, wrt: list = None, params: list
...
@@ -85,7 +85,7 @@ def bprop(func, *inputs, grads_wrt_outputs=None, wrt: list = None, params: list
if
not
params
:
if
not
params
:
params
=
func
.
trainable_params
()
params
=
func
.
trainable_params
()
grad_op
=
GradOperation
(
name
=
'grad'
,
get_all
=
wrt_inputs
,
get_by_list
=
wrt_params
,
sens_param
=
with_sens_param
)
grad_op
=
GradOperation
(
get_all
=
wrt_inputs
,
get_by_list
=
wrt_params
,
sens_param
=
with_sens_param
)
grad
=
Bprop
(
func
,
wrt_params
,
params
,
grad_op
,
grads_wrt_outputs
)
grad
=
Bprop
(
func
,
wrt_params
,
params
,
grad_op
,
grads_wrt_outputs
)
if
context
.
get_context
(
"mode"
)
==
context
.
PYNATIVE_MODE
:
if
context
.
get_context
(
"mode"
)
==
context
.
PYNATIVE_MODE
:
...
...
tests/mindspore_test_framework/utils/check_gradient.py
浏览文件 @
1a54785f
...
@@ -315,7 +315,7 @@ class ScalarGradChecker(_GradChecker):
...
@@ -315,7 +315,7 @@ class ScalarGradChecker(_GradChecker):
output_selector
=
None
,
output_selector
=
None
,
sampling_times
=-
1
,
sampling_times
=-
1
,
reduce_output
=
False
)
->
None
:
reduce_output
=
False
)
->
None
:
grad_op
=
GradOperation
(
'grad'
,
get_all
=
True
,
sens_param
=
True
)
grad_op
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
super
(
ScalarGradChecker
,
self
).
__init__
(
fn
,
grad_op
,
args
,
delta
,
max_error
,
input_selector
,
\
super
(
ScalarGradChecker
,
self
).
__init__
(
fn
,
grad_op
,
args
,
delta
,
max_error
,
input_selector
,
\
output_selector
,
sampling_times
,
reduce_output
)
output_selector
,
sampling_times
,
reduce_output
)
...
@@ -358,7 +358,7 @@ class OperationGradChecker(_GradChecker):
...
@@ -358,7 +358,7 @@ class OperationGradChecker(_GradChecker):
output_selector
=
None
,
output_selector
=
None
,
sampling_times
=-
1
,
sampling_times
=-
1
,
reduce_output
=
False
)
->
None
:
reduce_output
=
False
)
->
None
:
grad_op
=
GradOperation
(
'grad'
,
get_all
=
True
,
sens_param
=
True
)
grad_op
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
super
(
OperationGradChecker
,
self
).
__init__
(
fn
,
grad_op
,
args
,
delta
,
max_error
,
input_selector
,
\
super
(
OperationGradChecker
,
self
).
__init__
(
fn
,
grad_op
,
args
,
delta
,
max_error
,
input_selector
,
\
output_selector
,
sampling_times
,
reduce_output
)
output_selector
,
sampling_times
,
reduce_output
)
...
@@ -390,7 +390,7 @@ class NNGradChecker(_GradChecker):
...
@@ -390,7 +390,7 @@ class NNGradChecker(_GradChecker):
output_selector
=
None
,
output_selector
=
None
,
sampling_times
=-
1
,
sampling_times
=-
1
,
reduce_output
=
False
)
->
None
:
reduce_output
=
False
)
->
None
:
grad_op
=
GradOperation
(
'grad'
,
get_by_list
=
True
,
sens_param
=
True
)
grad_op
=
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
self
.
params
=
ParameterTuple
(
fn
.
trainable_params
())
self
.
params
=
ParameterTuple
(
fn
.
trainable_params
())
super
(
NNGradChecker
,
self
).
__init__
(
fn
,
grad_op
,
args
,
delta
,
max_error
,
input_selector
,
\
super
(
NNGradChecker
,
self
).
__init__
(
fn
,
grad_op
,
args
,
delta
,
max_error
,
input_selector
,
\
output_selector
,
sampling_times
,
reduce_output
)
output_selector
,
sampling_times
,
reduce_output
)
...
...
tests/ops_common.py
浏览文件 @
1a54785f
...
@@ -23,7 +23,7 @@ from mindspore import Tensor
...
@@ -23,7 +23,7 @@ from mindspore import Tensor
from
mindspore.common.api
import
_executor
from
mindspore.common.api
import
_executor
grad_all_with_sens
=
C
.
GradOperation
(
'grad_all_with_sens'
,
get_all
=
True
,
sens_param
=
True
)
grad_all_with_sens
=
C
.
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
class
InputBackward
(
nn
.
Cell
):
class
InputBackward
(
nn
.
Cell
):
...
...
tests/perf_test/test_lenet.py
浏览文件 @
1a54785f
...
@@ -27,7 +27,7 @@ from mindspore.common.api import _executor
...
@@ -27,7 +27,7 @@ from mindspore.common.api import _executor
context
.
set_context
(
mode
=
context
.
GRAPH_MODE
)
context
.
set_context
(
mode
=
context
.
GRAPH_MODE
)
grad_all_with_sens
=
C
.
GradOperation
(
'grad_all_with_sens'
,
get_all
=
True
,
sens_param
=
True
)
grad_all_with_sens
=
C
.
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
batch_size
=
1
batch_size
=
1
channel
=
1
channel
=
1
...
...
tests/st/control/test_cont_grad.py
浏览文件 @
1a54785f
...
@@ -28,8 +28,8 @@ from mindspore.ops import operations as P
...
@@ -28,8 +28,8 @@ from mindspore.ops import operations as P
# context.set_context(save_graphs=True)
# context.set_context(save_graphs=True)
grad_by_list
=
C
.
GradOperation
(
'get_by_list'
,
get_by_list
=
True
)
grad_by_list
=
C
.
GradOperation
(
get_by_list
=
True
)
grad_all
=
C
.
GradOperation
(
'get_all'
,
get_all
=
True
)
grad_all
=
C
.
GradOperation
(
get_all
=
True
)
def
test_while_forward
():
def
test_while_forward
():
...
...
tests/st/gnn/test_gnn_aggregator.py
浏览文件 @
1a54785f
...
@@ -25,7 +25,7 @@ from mindspore.common.api import _executor
...
@@ -25,7 +25,7 @@ from mindspore.common.api import _executor
context
.
set_context
(
mode
=
context
.
GRAPH_MODE
)
context
.
set_context
(
mode
=
context
.
GRAPH_MODE
)
grad_all_with_sens
=
C
.
GradOperation
(
'grad_all_with_sens'
,
get_all
=
True
,
sens_param
=
True
)
grad_all_with_sens
=
C
.
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
class
MeanAggregatorGrad
(
nn
.
Cell
):
class
MeanAggregatorGrad
(
nn
.
Cell
):
...
...
tests/st/model_zoo_tests/wide_and_deep/python_file_for_ci/wide_and_deep.py
浏览文件 @
1a54785f
...
@@ -284,9 +284,9 @@ class TrainStepWrap(nn.Cell):
...
@@ -284,9 +284,9 @@ class TrainStepWrap(nn.Cell):
self
.
optimizer_d
=
Adam
(
self
.
optimizer_d
=
Adam
(
self
.
weights_d
,
learning_rate
=
3.5e-4
,
eps
=
1e-8
,
loss_scale
=
sens
)
self
.
weights_d
,
learning_rate
=
3.5e-4
,
eps
=
1e-8
,
loss_scale
=
sens
)
self
.
hyper_map
=
C
.
HyperMap
()
self
.
hyper_map
=
C
.
HyperMap
()
self
.
grad_w
=
C
.
GradOperation
(
'grad_w'
,
get_by_list
=
True
,
self
.
grad_w
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
sens_param
=
True
)
self
.
grad_d
=
C
.
GradOperation
(
'grad_d'
,
get_by_list
=
True
,
self
.
grad_d
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
sens_param
=
True
)
self
.
sens
=
sens
self
.
sens
=
sens
self
.
loss_net_w
=
IthOutputCell
(
network
,
output_index
=
0
)
self
.
loss_net_w
=
IthOutputCell
(
network
,
output_index
=
0
)
...
...
tests/st/model_zoo_tests/yolov3/src/yolov3.py
浏览文件 @
1a54785f
...
@@ -647,7 +647,7 @@ class TrainingWrapper(nn.Cell):
...
@@ -647,7 +647,7 @@ class TrainingWrapper(nn.Cell):
self
.
network
=
network
self
.
network
=
network
self
.
weights
=
ms
.
ParameterTuple
(
network
.
trainable_params
())
self
.
weights
=
ms
.
ParameterTuple
(
network
.
trainable_params
())
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
get_by_list
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
self
.
sens
=
sens
self
.
sens
=
sens
self
.
reducer_flag
=
False
self
.
reducer_flag
=
False
self
.
grad_reducer
=
None
self
.
grad_reducer
=
None
...
...
tests/st/networks/models/bert/src/bert_for_pre_training.py
浏览文件 @
1a54785f
...
@@ -271,7 +271,7 @@ class BertTrainOneStepCell(nn.Cell):
...
@@ -271,7 +271,7 @@ class BertTrainOneStepCell(nn.Cell):
self
.
network
=
network
self
.
network
=
network
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
get_by_list
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
self
.
sens
=
sens
self
.
sens
=
sens
self
.
reducer_flag
=
False
self
.
reducer_flag
=
False
self
.
parallel_mode
=
context
.
get_auto_parallel_context
(
"parallel_mode"
)
self
.
parallel_mode
=
context
.
get_auto_parallel_context
(
"parallel_mode"
)
...
@@ -351,8 +351,7 @@ class BertTrainOneStepWithLossScaleCell(nn.Cell):
...
@@ -351,8 +351,7 @@ class BertTrainOneStepWithLossScaleCell(nn.Cell):
self
.
network
=
network
self
.
network
=
network
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
get_by_list
=
True
,
sens_param
=
True
)
sens_param
=
True
)
self
.
reducer_flag
=
False
self
.
reducer_flag
=
False
self
.
allreduce
=
P
.
AllReduce
()
self
.
allreduce
=
P
.
AllReduce
()
...
...
tests/st/networks/models/bert/src/utils.py
浏览文件 @
1a54785f
...
@@ -52,8 +52,7 @@ class BertFinetuneCell(nn.Cell):
...
@@ -52,8 +52,7 @@ class BertFinetuneCell(nn.Cell):
self
.
network
=
network
self
.
network
=
network
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
get_by_list
=
True
,
sens_param
=
True
)
sens_param
=
True
)
self
.
reducer_flag
=
False
self
.
reducer_flag
=
False
self
.
allreduce
=
P
.
AllReduce
()
self
.
allreduce
=
P
.
AllReduce
()
...
...
tests/st/networks/models/resnet50/src_thor/grad_reducer_thor.py
浏览文件 @
1a54785f
...
@@ -120,7 +120,7 @@ class DistributedGradReducerThor(Cell):
...
@@ -120,7 +120,7 @@ class DistributedGradReducerThor(Cell):
>>> self.network.add_flags(defer_inline=True)
>>> self.network.add_flags(defer_inline=True)
>>> self.weights = ParameterTuple(network.trainable_params())
>>> self.weights = ParameterTuple(network.trainable_params())
>>> self.optimizer = optimizer
>>> self.optimizer = optimizer
>>> self.grad = C.GradOperation(
'grad',
get_by_list=True, sens_param=True)
>>> self.grad = C.GradOperation(get_by_list=True, sens_param=True)
>>> self.sens = sens
>>> self.sens = sens
>>> self.reducer_flag = False
>>> self.reducer_flag = False
>>> self.grad_reducer = None
>>> self.grad_reducer = None
...
...
tests/st/networks/test_cell_bprop.py
浏览文件 @
1a54785f
...
@@ -29,7 +29,7 @@ from mindspore.ops import operations as P
...
@@ -29,7 +29,7 @@ from mindspore.ops import operations as P
context
.
set_context
(
mode
=
context
.
GRAPH_MODE
,
device_target
=
"Ascend"
)
context
.
set_context
(
mode
=
context
.
GRAPH_MODE
,
device_target
=
"Ascend"
)
grad_all
=
C
.
GradOperation
(
'get_all'
,
get_all
=
True
)
grad_all
=
C
.
GradOperation
(
get_all
=
True
)
class
MulAdd
(
nn
.
Cell
):
class
MulAdd
(
nn
.
Cell
):
...
@@ -351,7 +351,7 @@ class MulAddWithParam(nn.Cell):
...
@@ -351,7 +351,7 @@ class MulAddWithParam(nn.Cell):
@
pytest
.
mark
.
platform_x86_ascend_training
@
pytest
.
mark
.
platform_x86_ascend_training
@
pytest
.
mark
.
env_onecard
@
pytest
.
mark
.
env_onecard
def
test_refkey_bprop
():
def
test_refkey_bprop
():
grad_by_list
=
C
.
GradOperation
(
'get_by_list'
,
get_all
=
True
,
get_by_list
=
True
)
grad_by_list
=
C
.
GradOperation
(
get_all
=
True
,
get_by_list
=
True
)
class
GradWrap
(
nn
.
Cell
):
class
GradWrap
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
GradWrap
,
self
).
__init__
()
super
(
GradWrap
,
self
).
__init__
()
...
...
tests/st/ops/ascend/test_addn.py
浏览文件 @
1a54785f
...
@@ -49,7 +49,7 @@ def test_net():
...
@@ -49,7 +49,7 @@ def test_net():
def
test_grad_addn_with_list
():
def
test_grad_addn_with_list
():
grad_op
=
C
.
GradOperation
(
'get_all'
,
get_all
=
True
)
grad_op
=
C
.
GradOperation
(
get_all
=
True
)
class
AddN
(
nn
.
Cell
):
class
AddN
(
nn
.
Cell
):
def
__init__
(
self
):
def
__init__
(
self
):
super
().
__init__
()
super
().
__init__
()
...
...
tests/st/ops/ascend/test_conv_grad.py
浏览文件 @
1a54785f
...
@@ -29,7 +29,7 @@ context.set_context(device_target="Ascend")
...
@@ -29,7 +29,7 @@ context.set_context(device_target="Ascend")
class
Grad
(
nn
.
Cell
):
class
Grad
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
@
ms_function
@
ms_function
...
...
tests/st/ops/ascend/test_dense_grad.py
浏览文件 @
1a54785f
...
@@ -26,7 +26,7 @@ context.set_context(device_target="Ascend")
...
@@ -26,7 +26,7 @@ context.set_context(device_target="Ascend")
class
Grad
(
nn
.
Cell
):
class
Grad
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
@
ms_function
@
ms_function
...
...
tests/st/ops/ascend/test_fused_batchnorm_grad.py
浏览文件 @
1a54785f
...
@@ -30,7 +30,7 @@ context.set_context(device_target="Ascend")
...
@@ -30,7 +30,7 @@ context.set_context(device_target="Ascend")
class
Grad
(
nn
.
Cell
):
class
Grad
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
@
ms_function
@
ms_function
...
...
tests/st/ops/ascend/test_maxpool_grad.py
浏览文件 @
1a54785f
...
@@ -27,7 +27,7 @@ context.set_context(device_target="Ascend")
...
@@ -27,7 +27,7 @@ context.set_context(device_target="Ascend")
class
Grad
(
nn
.
Cell
):
class
Grad
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
@
ms_function
@
ms_function
...
...
tests/st/ops/ascend/test_maxpool_with_argmax_grad.py
浏览文件 @
1a54785f
...
@@ -27,7 +27,7 @@ context.set_context(device_target="Ascend")
...
@@ -27,7 +27,7 @@ context.set_context(device_target="Ascend")
class
Grad
(
nn
.
Cell
):
class
Grad
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
@
ms_function
@
ms_function
...
...
tests/st/ops/ascend/test_relu_grad.py
浏览文件 @
1a54785f
...
@@ -27,7 +27,7 @@ context.set_context(device_target="Ascend")
...
@@ -27,7 +27,7 @@ context.set_context(device_target="Ascend")
class
Grad
(
nn
.
Cell
):
class
Grad
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
@
ms_function
@
ms_function
...
...
tests/st/ops/ascend/test_simplemean_grad.py
浏览文件 @
1a54785f
...
@@ -27,7 +27,7 @@ context.set_context(device_target="Ascend")
...
@@ -27,7 +27,7 @@ context.set_context(device_target="Ascend")
class
Grad
(
nn
.
Cell
):
class
Grad
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
@
ms_function
@
ms_function
...
...
tests/st/ops/ascend/test_tbe_ops/test_batchnorm_grad.py
浏览文件 @
1a54785f
...
@@ -30,7 +30,7 @@ context.set_context(device_target="Ascend")
...
@@ -30,7 +30,7 @@ context.set_context(device_target="Ascend")
class
Grad
(
nn
.
Cell
):
class
Grad
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
@
ms_function
@
ms_function
...
...
tests/st/ops/ascend/test_tbe_ops/test_gelu_grad_sens.py
浏览文件 @
1a54785f
...
@@ -27,7 +27,7 @@ context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
...
@@ -27,7 +27,7 @@ context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
class
Grad
(
Cell
):
class
Grad
(
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
def
construct
(
self
,
input_
,
output_grad
):
def
construct
(
self
,
input_
,
output_grad
):
...
@@ -71,7 +71,7 @@ class MEGeluLargeIn(Cell):
...
@@ -71,7 +71,7 @@ class MEGeluLargeIn(Cell):
class
GradLargeIn
(
Cell
):
class
GradLargeIn
(
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
GradLargeIn
,
self
).
__init__
()
super
(
GradLargeIn
,
self
).
__init__
()
self
.
grad
=
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
def
construct
(
self
,
x1
,
x2
,
output_grad
):
def
construct
(
self
,
x1
,
x2
,
output_grad
):
...
...
tests/st/ops/ascend/test_tbe_ops/test_layernorm_grad.py
浏览文件 @
1a54785f
...
@@ -27,7 +27,7 @@ context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
...
@@ -27,7 +27,7 @@ context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
class
Grad
(
Cell
):
class
Grad
(
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
def
construct
(
self
,
input_
,
output_grad
,):
def
construct
(
self
,
input_
,
output_grad
,):
...
...
tests/st/ops/ascend/test_tbe_ops/test_maximum_grad.py
浏览文件 @
1a54785f
...
@@ -21,7 +21,7 @@ from mindspore.ops import composite as C
...
@@ -21,7 +21,7 @@ from mindspore.ops import composite as C
from
mindspore.ops
import
operations
as
P
from
mindspore.ops
import
operations
as
P
context
.
set_context
(
device_target
=
"Ascend"
)
context
.
set_context
(
device_target
=
"Ascend"
)
grad
=
C
.
GradOperation
(
'get_all'
,
get_all
=
True
,
sens_param
=
True
)
grad
=
C
.
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
class
MaxNetMe
(
Cell
):
class
MaxNetMe
(
Cell
):
...
...
tests/st/ops/ascend/test_tbe_ops/test_maxpool_grad.py
浏览文件 @
1a54785f
...
@@ -27,7 +27,7 @@ context.set_context(device_target="Ascend")
...
@@ -27,7 +27,7 @@ context.set_context(device_target="Ascend")
class
Grad
(
nn
.
Cell
):
class
Grad
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
@
ms_function
@
ms_function
...
...
tests/st/ops/ascend/test_tbe_ops/test_minimum_grad.py
浏览文件 @
1a54785f
...
@@ -21,7 +21,7 @@ from mindspore.ops import composite as C
...
@@ -21,7 +21,7 @@ from mindspore.ops import composite as C
from
mindspore.ops.operations
import
Minimum
from
mindspore.ops.operations
import
Minimum
context
.
set_context
(
mode
=
context
.
GRAPH_MODE
,
device_target
=
"Ascend"
)
context
.
set_context
(
mode
=
context
.
GRAPH_MODE
,
device_target
=
"Ascend"
)
grad
=
C
.
GradOperation
(
'get_all'
,
get_all
=
True
,
sens_param
=
True
)
grad
=
C
.
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
class
MinNetMe
(
Cell
):
class
MinNetMe
(
Cell
):
...
...
tests/st/ops/ascend/test_tbe_ops/test_relu_grad.py
浏览文件 @
1a54785f
...
@@ -27,7 +27,7 @@ context.set_context(device_target="Ascend")
...
@@ -27,7 +27,7 @@ context.set_context(device_target="Ascend")
class
Grad
(
nn
.
Cell
):
class
Grad
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
@
ms_function
@
ms_function
...
...
tests/st/ops/ascend/test_tbe_ops/test_relu_v2_grad.py
浏览文件 @
1a54785f
...
@@ -27,7 +27,7 @@ context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
...
@@ -27,7 +27,7 @@ context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
class
Grad
(
nn
.
Cell
):
class
Grad
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
GradOperation
(
name
=
"get_all"
,
get_all
=
True
)
self
.
grad
=
GradOperation
(
get_all
=
True
)
self
.
network
=
network
self
.
network
=
network
@
ms_function
@
ms_function
...
...
tests/st/ops/ascend/test_tbe_ops/test_resize_nearest_neighbor_grad.py
浏览文件 @
1a54785f
...
@@ -37,7 +37,7 @@ class Net(nn.Cell):
...
@@ -37,7 +37,7 @@ class Net(nn.Cell):
class
Grad
(
nn
.
Cell
):
class
Grad
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
@
ms_function
@
ms_function
...
...
tests/st/ops/ascend/test_tbe_ops/test_sigmoid_cross_entropy_with_logits_grad.py
浏览文件 @
1a54785f
...
@@ -37,7 +37,7 @@ class Net(nn.Cell):
...
@@ -37,7 +37,7 @@ class Net(nn.Cell):
class
Grad
(
nn
.
Cell
):
class
Grad
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
@
ms_function
@
ms_function
...
...
tests/st/ops/ascend/test_tbe_ops/test_sigmoid_grad.py
浏览文件 @
1a54785f
...
@@ -37,7 +37,7 @@ class Net(nn.Cell):
...
@@ -37,7 +37,7 @@ class Net(nn.Cell):
class
Grad
(
nn
.
Cell
):
class
Grad
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
@
ms_function
@
ms_function
...
...
tests/st/ops/ascend/test_tbe_ops/test_smooth_l1_loss_grad.py
浏览文件 @
1a54785f
...
@@ -36,7 +36,7 @@ class Net(nn.Cell):
...
@@ -36,7 +36,7 @@ class Net(nn.Cell):
class
Grad
(
nn
.
Cell
):
class
Grad
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
def
construct
(
self
,
pred
,
gt
,
dout
):
def
construct
(
self
,
pred
,
gt
,
dout
):
...
...
tests/st/ops/ascend/test_tbe_ops/test_stridedslice_grad.py
浏览文件 @
1a54785f
...
@@ -26,7 +26,7 @@ context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
...
@@ -26,7 +26,7 @@ context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
class
Grad
(
Cell
):
class
Grad
(
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
def
construct
(
self
,
input_
,
output_grad
):
def
construct
(
self
,
input_
,
output_grad
):
...
...
tests/st/ops/cpu/test_batchnorm_op.py
浏览文件 @
1a54785f
...
@@ -37,7 +37,7 @@ class Batchnorm_Net(Cell):
...
@@ -37,7 +37,7 @@ class Batchnorm_Net(Cell):
class
Grad
(
Cell
):
class
Grad
(
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
C
.
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
def
construct
(
self
,
input_data
,
sens
):
def
construct
(
self
,
input_data
,
sens
):
...
...
tests/st/ops/cpu/test_lstm_op.py
浏览文件 @
1a54785f
...
@@ -207,8 +207,7 @@ class Grad(nn.Cell):
...
@@ -207,8 +207,7 @@ class Grad(nn.Cell):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
network
=
network
self
.
network
=
network
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
grad
=
C
.
GradOperation
(
'grad'
,
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
get_by_list
=
True
,
sens_param
=
True
)
sens_param
=
True
)
@
ms_function
@
ms_function
...
...
tests/st/ops/custom_ops_tbe/test_square.py
浏览文件 @
1a54785f
...
@@ -23,7 +23,7 @@ from mindspore.ops import composite as C
...
@@ -23,7 +23,7 @@ from mindspore.ops import composite as C
context
.
set_context
(
mode
=
context
.
GRAPH_MODE
,
device_target
=
"Ascend"
)
context
.
set_context
(
mode
=
context
.
GRAPH_MODE
,
device_target
=
"Ascend"
)
grad_with_sens
=
C
.
GradOperation
(
'grad_with_sens'
,
sens_param
=
True
)
grad_with_sens
=
C
.
GradOperation
(
sens_param
=
True
)
class
Net
(
nn
.
Cell
):
class
Net
(
nn
.
Cell
):
...
...
tests/st/ops/gpu/test_batchnorm_op.py
浏览文件 @
1a54785f
...
@@ -37,7 +37,7 @@ class Batchnorm_Net(Cell):
...
@@ -37,7 +37,7 @@ class Batchnorm_Net(Cell):
class
Grad
(
Cell
):
class
Grad
(
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
C
.
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
def
construct
(
self
,
input_data
,
sens
):
def
construct
(
self
,
input_data
,
sens
):
...
...
tests/st/ops/gpu/test_binary_cross_entropy_op.py
浏览文件 @
1a54785f
...
@@ -54,7 +54,7 @@ def test_binary_cross_entropy_loss():
...
@@ -54,7 +54,7 @@ def test_binary_cross_entropy_loss():
class
Grad
(
nn
.
Cell
):
class
Grad
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
C
.
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
def
construct
(
self
,
x1
,
x2
,
sens
,
weight
):
def
construct
(
self
,
x1
,
x2
,
sens
,
weight
):
...
...
tests/st/ops/gpu/test_ctcloss_op.py
浏览文件 @
1a54785f
...
@@ -40,7 +40,7 @@ class Net(nn.Cell):
...
@@ -40,7 +40,7 @@ class Net(nn.Cell):
class
GradData
(
nn
.
Cell
):
class
GradData
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
GradData
,
self
).
__init__
()
super
(
GradData
,
self
).
__init__
()
self
.
grad
=
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
False
)
self
.
grad
=
GradOperation
(
get_all
=
True
,
sens_param
=
False
)
self
.
network
=
network
self
.
network
=
network
def
construct
(
self
,
probs
,
labels
,
input_lengths
,
label_lengths
):
def
construct
(
self
,
probs
,
labels
,
input_lengths
,
label_lengths
):
...
...
tests/st/ops/gpu/test_dense_op.py
浏览文件 @
1a54785f
...
@@ -65,7 +65,7 @@ def test_biasadd():
...
@@ -65,7 +65,7 @@ def test_biasadd():
class
GradData
(
nn
.
Cell
):
class
GradData
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
GradData
,
self
).
__init__
()
super
(
GradData
,
self
).
__init__
()
self
.
grad
=
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
def
construct
(
self
,
inputs
,
output_grad
):
def
construct
(
self
,
inputs
,
output_grad
):
...
@@ -77,8 +77,7 @@ class GradWeight(nn.Cell):
...
@@ -77,8 +77,7 @@ class GradWeight(nn.Cell):
super
(
GradWeight
,
self
).
__init__
()
super
(
GradWeight
,
self
).
__init__
()
self
.
network
=
network
self
.
network
=
network
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
grad
=
C
.
GradOperation
(
'grad'
,
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
get_by_list
=
True
,
sens_param
=
True
)
sens_param
=
True
)
def
construct
(
self
,
x
,
output_grad
):
def
construct
(
self
,
x
,
output_grad
):
...
@@ -169,7 +168,7 @@ def test_dw():
...
@@ -169,7 +168,7 @@ def test_dw():
class
Grad
(
nn
.
Cell
):
class
Grad
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
def
construct
(
self
,
input_
,
bias
,
dy
):
def
construct
(
self
,
input_
,
bias
,
dy
):
...
...
tests/st/ops/gpu/test_gelu_grad_op.py
浏览文件 @
1a54785f
...
@@ -37,7 +37,7 @@ class GeluNet(nn.Cell):
...
@@ -37,7 +37,7 @@ class GeluNet(nn.Cell):
class
Grad
(
nn
.
Cell
):
class
Grad
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
C
.
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
def
construct
(
self
,
input_data
,
sens
):
def
construct
(
self
,
input_data
,
sens
):
...
...
tests/st/ops/gpu/test_kl_div_op.py
浏览文件 @
1a54785f
...
@@ -53,7 +53,7 @@ def test_binary_cross_entropy_loss():
...
@@ -53,7 +53,7 @@ def test_binary_cross_entropy_loss():
class
Grad
(
nn
.
Cell
):
class
Grad
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
C
.
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
def
construct
(
self
,
x1
,
x2
,
sens
):
def
construct
(
self
,
x1
,
x2
,
sens
):
...
...
tests/st/ops/gpu/test_logsoftmax_op.py
浏览文件 @
1a54785f
...
@@ -52,7 +52,7 @@ class LogSoftmax(nn.Cell):
...
@@ -52,7 +52,7 @@ class LogSoftmax(nn.Cell):
class
Grad
(
nn
.
Cell
):
class
Grad
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
C
.
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
def
construct
(
self
,
input_data
,
sens
):
def
construct
(
self
,
input_data
,
sens
):
...
...
tests/st/ops/gpu/test_lstm_op.py
浏览文件 @
1a54785f
...
@@ -581,8 +581,7 @@ class Grad(nn.Cell):
...
@@ -581,8 +581,7 @@ class Grad(nn.Cell):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
network
=
network
self
.
network
=
network
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
grad
=
C
.
GradOperation
(
'grad'
,
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
get_by_list
=
True
,
sens_param
=
True
)
sens_param
=
True
)
@
ms_function
@
ms_function
...
...
tests/st/ops/gpu/test_maximum_op.py
浏览文件 @
1a54785f
...
@@ -35,7 +35,7 @@ class Net(Cell):
...
@@ -35,7 +35,7 @@ class Net(Cell):
class
Grad
(
Cell
):
class
Grad
(
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
C
.
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
def
construct
(
self
,
x1
,
x2
,
sens
):
def
construct
(
self
,
x1
,
x2
,
sens
):
...
...
tests/st/ops/gpu/test_minimum_op.py
浏览文件 @
1a54785f
...
@@ -36,7 +36,7 @@ class MinimumNet(Cell):
...
@@ -36,7 +36,7 @@ class MinimumNet(Cell):
class
Grad
(
Cell
):
class
Grad
(
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
C
.
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
def
construct
(
self
,
x1
,
x2
,
sens
):
def
construct
(
self
,
x1
,
x2
,
sens
):
...
...
tests/st/ops/gpu/test_mirror_pad.py
浏览文件 @
1a54785f
...
@@ -58,7 +58,7 @@ def test_mirror_pad():
...
@@ -58,7 +58,7 @@ def test_mirror_pad():
class
Grad
(
nn
.
Cell
):
class
Grad
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
def
construct
(
self
,
input_
,
output_grad
):
def
construct
(
self
,
input_
,
output_grad
):
return
self
.
grad
(
self
.
network
)(
input_
,
output_grad
)
return
self
.
grad
(
self
.
network
)(
input_
,
output_grad
)
...
...
tests/st/ops/gpu/test_smoothl1loss_op.py
浏览文件 @
1a54785f
...
@@ -59,7 +59,7 @@ def test_smoothl1loss():
...
@@ -59,7 +59,7 @@ def test_smoothl1loss():
class
Grad
(
nn
.
Cell
):
class
Grad
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
C
.
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
def
construct
(
self
,
x1
,
x2
,
sens
):
def
construct
(
self
,
x1
,
x2
,
sens
):
...
...
tests/st/ops/gpu/test_softmax_op.py
浏览文件 @
1a54785f
...
@@ -79,7 +79,7 @@ class Net(nn.Cell):
...
@@ -79,7 +79,7 @@ class Net(nn.Cell):
class
Grad
(
nn
.
Cell
):
class
Grad
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
C
.
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
def
construct
(
self
,
input_data
,
sens
):
def
construct
(
self
,
input_data
,
sens
):
...
...
tests/st/ops/gpu/test_stridedslice_grad_op.py
浏览文件 @
1a54785f
...
@@ -36,7 +36,7 @@ class StridedSliceNet(nn.Cell):
...
@@ -36,7 +36,7 @@ class StridedSliceNet(nn.Cell):
class
GradData
(
nn
.
Cell
):
class
GradData
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
GradData
,
self
).
__init__
()
super
(
GradData
,
self
).
__init__
()
self
.
grad
=
C
.
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
False
)
self
.
grad
=
C
.
GradOperation
(
get_all
=
True
,
sens_param
=
False
)
self
.
network
=
network
self
.
network
=
network
def
construct
(
self
,
x
):
def
construct
(
self
,
x
):
...
...
tests/st/ops/gpu/test_tanh_op.py
浏览文件 @
1a54785f
...
@@ -37,7 +37,7 @@ class TanhNet(nn.Cell):
...
@@ -37,7 +37,7 @@ class TanhNet(nn.Cell):
class
Grad
(
nn
.
Cell
):
class
Grad
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
C
.
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
def
construct
(
self
,
input_data
,
sens
):
def
construct
(
self
,
input_data
,
sens
):
...
...
tests/st/pynative/test_pynative_hook.py
浏览文件 @
1a54785f
...
@@ -30,7 +30,7 @@ from mindspore.common.initializer import TruncatedNormal
...
@@ -30,7 +30,7 @@ from mindspore.common.initializer import TruncatedNormal
context
.
set_context
(
mode
=
context
.
PYNATIVE_MODE
,
device_target
=
"Ascend"
)
context
.
set_context
(
mode
=
context
.
PYNATIVE_MODE
,
device_target
=
"Ascend"
)
grad_all
=
C
.
GradOperation
(
'get_all'
,
get_all
=
True
)
grad_all
=
C
.
GradOperation
(
get_all
=
True
)
def
weight_variable
():
def
weight_variable
():
...
@@ -112,7 +112,7 @@ class GradWrap(nn.Cell):
...
@@ -112,7 +112,7 @@ class GradWrap(nn.Cell):
def
construct
(
self
,
x
,
label
):
def
construct
(
self
,
x
,
label
):
weights
=
self
.
weights
weights
=
self
.
weights
return
C
.
GradOperation
(
'get_by_list'
,
get_by_list
=
True
)(
self
.
network
,
weights
)(
x
,
label
)
return
C
.
GradOperation
(
get_by_list
=
True
)(
self
.
network
,
weights
)(
x
,
label
)
class
test_custom_cell_base
():
class
test_custom_cell_base
():
...
...
tests/st/pynative/test_pynative_lenet.py
浏览文件 @
1a54785f
...
@@ -29,7 +29,7 @@ from mindspore.ops import operations as P
...
@@ -29,7 +29,7 @@ from mindspore.ops import operations as P
np
.
random
.
seed
(
1
)
np
.
random
.
seed
(
1
)
grad_by_list
=
C
.
GradOperation
(
'get_by_list'
,
get_by_list
=
True
)
grad_by_list
=
C
.
GradOperation
(
get_by_list
=
True
)
def
weight_variable
():
def
weight_variable
():
...
...
tests/st/pynative/test_pynative_mindarmour.py
浏览文件 @
1a54785f
...
@@ -87,7 +87,7 @@ class LeNet(nn.Cell):
...
@@ -87,7 +87,7 @@ class LeNet(nn.Cell):
class
GradWithSens
(
Cell
):
class
GradWithSens
(
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
GradWithSens
,
self
).
__init__
()
super
(
GradWithSens
,
self
).
__init__
()
self
.
grad
=
GradOperation
(
name
=
"grad"
,
get_all
=
False
,
self
.
grad
=
GradOperation
(
get_all
=
False
,
sens_param
=
True
)
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
...
@@ -99,8 +99,7 @@ class GradWithSens(Cell):
...
@@ -99,8 +99,7 @@ class GradWithSens(Cell):
class
GradWrapWithLoss
(
Cell
):
class
GradWrapWithLoss
(
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
GradWrapWithLoss
,
self
).
__init__
()
super
(
GradWrapWithLoss
,
self
).
__init__
()
self
.
_grad_all
=
GradOperation
(
name
=
"get_all"
,
self
.
_grad_all
=
GradOperation
(
get_all
=
True
,
get_all
=
True
,
sens_param
=
False
)
sens_param
=
False
)
self
.
_network
=
network
self
.
_network
=
network
...
...
tests/st/pynative/test_pynative_resnet50.py
浏览文件 @
1a54785f
...
@@ -40,7 +40,7 @@ np.random.seed(1)
...
@@ -40,7 +40,7 @@ np.random.seed(1)
ds
.
config
.
set_seed
(
1
)
ds
.
config
.
set_seed
(
1
)
grad_by_list
=
CP
.
GradOperation
(
'get_by_list'
,
get_by_list
=
True
)
grad_by_list
=
CP
.
GradOperation
(
get_by_list
=
True
)
def
weight_variable
(
shape
):
def
weight_variable
(
shape
):
...
...
tests/st/pynative/test_tensor_index.py
浏览文件 @
1a54785f
...
@@ -24,7 +24,7 @@ from mindspore.common.parameter import ParameterTuple
...
@@ -24,7 +24,7 @@ from mindspore.common.parameter import ParameterTuple
from
mindspore.ops
import
composite
as
C
from
mindspore.ops
import
composite
as
C
grad_by_list_with_sens
=
C
.
GradOperation
(
'grad_by_list_with_sens'
,
get_by_list
=
True
,
sens_param
=
True
)
grad_by_list_with_sens
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
def
setup_module
():
def
setup_module
():
...
...
tests/train_step_wrap.py
浏览文件 @
1a54785f
...
@@ -32,7 +32,7 @@ class TrainStepWrap(nn.Cell):
...
@@ -32,7 +32,7 @@ class TrainStepWrap(nn.Cell):
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
optimizer
=
nn
.
Momentum
(
self
.
weights
,
0.1
,
0.9
)
self
.
optimizer
=
nn
.
Momentum
(
self
.
weights
,
0.1
,
0.9
)
self
.
hyper_map
=
C
.
HyperMap
()
self
.
hyper_map
=
C
.
HyperMap
()
self
.
grad
=
C
.
GradOperation
(
'grad'
,
get_by_list
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
)
def
construct
(
self
,
x
,
label
):
def
construct
(
self
,
x
,
label
):
weights
=
self
.
weights
weights
=
self
.
weights
...
@@ -71,7 +71,7 @@ class TrainStepWrap2(nn.Cell):
...
@@ -71,7 +71,7 @@ class TrainStepWrap2(nn.Cell):
self
.
weights
=
ParameterTuple
(
network
.
get_parameters
())
self
.
weights
=
ParameterTuple
(
network
.
get_parameters
())
self
.
optimizer
=
nn
.
Momentum
(
self
.
weights
,
0.1
,
0.9
)
self
.
optimizer
=
nn
.
Momentum
(
self
.
weights
,
0.1
,
0.9
)
self
.
hyper_map
=
C
.
HyperMap
()
self
.
hyper_map
=
C
.
HyperMap
()
self
.
grad
=
C
.
GradOperation
(
'grad'
,
get_by_list
=
True
,
sens_param
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
self
.
sens
=
sens
self
.
sens
=
sens
def
construct
(
self
,
x
):
def
construct
(
self
,
x
):
...
@@ -93,7 +93,7 @@ class TrainStepWrapWithoutOpt(nn.Cell):
...
@@ -93,7 +93,7 @@ class TrainStepWrapWithoutOpt(nn.Cell):
super
(
TrainStepWrapWithoutOpt
,
self
).
__init__
()
super
(
TrainStepWrapWithoutOpt
,
self
).
__init__
()
self
.
network
=
network
self
.
network
=
network
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
grad
=
C
.
GradOperation
(
'grad'
,
get_by_list
=
True
)
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
)
def
construct
(
self
,
x
,
label
):
def
construct
(
self
,
x
,
label
):
grads
=
self
.
grad
(
self
.
network
,
self
.
weights
)(
x
,
label
)
grads
=
self
.
grad
(
self
.
network
,
self
.
weights
)(
x
,
label
)
...
...
tests/ut/python/dtype/test_list.py
浏览文件 @
1a54785f
...
@@ -31,7 +31,7 @@ from tests.mindspore_test_framework.pipeline.forward.compile_forward \
...
@@ -31,7 +31,7 @@ from tests.mindspore_test_framework.pipeline.forward.compile_forward \
context
.
set_context
(
mode
=
context
.
GRAPH_MODE
)
context
.
set_context
(
mode
=
context
.
GRAPH_MODE
)
grad_all
=
C
.
GradOperation
(
'get_all'
,
get_all
=
True
)
grad_all
=
C
.
GradOperation
(
get_all
=
True
)
def
test_list_equal
():
def
test_list_equal
():
...
...
tests/ut/python/exec/test_train_with_lars.py
浏览文件 @
1a54785f
...
@@ -52,8 +52,7 @@ class TrainOneStepWithLarsCell(nn.Cell):
...
@@ -52,8 +52,7 @@ class TrainOneStepWithLarsCell(nn.Cell):
self
.
slice_index
,
self
.
params_len
,
weights
=
get_net_trainable_reordered_params
(
self
.
network
)
self
.
slice_index
,
self
.
params_len
,
weights
=
get_net_trainable_reordered_params
(
self
.
network
)
self
.
weights
=
ParameterTuple
(
weights
)
self
.
weights
=
ParameterTuple
(
weights
)
self
.
optimizer
=
optimizer
self
.
optimizer
=
optimizer
self
.
grad
=
C
.
GradOperation
(
'grad'
,
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
get_by_list
=
True
,
sens_param
=
True
)
sens_param
=
True
)
self
.
sens
=
Parameter
(
Tensor
([
sens
],
mstype
.
float32
),
name
=
'sens'
,
requires_grad
=
False
)
self
.
sens
=
Parameter
(
Tensor
([
sens
],
mstype
.
float32
),
name
=
'sens'
,
requires_grad
=
False
)
self
.
weight_decay
=
1.0
self
.
weight_decay
=
1.0
...
...
tests/ut/python/ir/test_row_tensor.py
浏览文件 @
1a54785f
...
@@ -248,7 +248,7 @@ def test_row_tensor_attr():
...
@@ -248,7 +248,7 @@ def test_row_tensor_attr():
def
test_row_tensor_sparse_gatherv2_grad_all
():
def
test_row_tensor_sparse_gatherv2_grad_all
():
grad_all
=
C
.
GradOperation
(
'get_all'
,
get_all
=
True
)
grad_all
=
C
.
GradOperation
(
get_all
=
True
)
class
GradWrap
(
nn
.
Cell
):
class
GradWrap
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
GradWrap
,
self
).
__init__
()
super
(
GradWrap
,
self
).
__init__
()
...
@@ -269,7 +269,7 @@ def test_row_tensor_sparse_gatherv2_grad_all():
...
@@ -269,7 +269,7 @@ def test_row_tensor_sparse_gatherv2_grad_all():
def
test_row_tensor_sparse_gatherv2_grad_with_pram
():
def
test_row_tensor_sparse_gatherv2_grad_with_pram
():
grad_by_list
=
C
.
GradOperation
(
'get_by_list'
,
get_by_list
=
True
)
grad_by_list
=
C
.
GradOperation
(
get_by_list
=
True
)
class
GradWrap
(
nn
.
Cell
):
class
GradWrap
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
GradWrap
,
self
).
__init__
()
super
(
GradWrap
,
self
).
__init__
()
...
...
tests/ut/python/ir/test_sparse_tensor.py
浏览文件 @
1a54785f
...
@@ -28,7 +28,7 @@ from mindspore import Tensor, SparseTensor, context
...
@@ -28,7 +28,7 @@ from mindspore import Tensor, SparseTensor, context
context
.
set_context
(
mode
=
context
.
GRAPH_MODE
,
enable_sparse
=
True
)
context
.
set_context
(
mode
=
context
.
GRAPH_MODE
,
enable_sparse
=
True
)
grad_op
=
C
.
GradOperation
(
'get_all'
,
get_all
=
True
)
grad_op
=
C
.
GradOperation
(
get_all
=
True
)
class
MakeSparseTensor
(
nn
.
Cell
):
class
MakeSparseTensor
(
nn
.
Cell
):
def
__init__
(
self
,
dense_shape
):
def
__init__
(
self
,
dense_shape
):
...
...
tests/ut/python/keep_order/test_keep_order.py
浏览文件 @
1a54785f
...
@@ -50,7 +50,7 @@ class Func(nn.Cell):
...
@@ -50,7 +50,7 @@ class Func(nn.Cell):
return
out
return
out
grad_s
=
C
.
GradOperation
(
'grad_with_sens'
,
get_all
=
True
,
sens_param
=
True
)
grad_s
=
C
.
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
class
Net
(
nn
.
Cell
):
class
Net
(
nn
.
Cell
):
...
...
tests/ut/python/model/test_mix_precision.py
浏览文件 @
1a54785f
...
@@ -166,8 +166,7 @@ class GetParamGrad(nn.Cell):
...
@@ -166,8 +166,7 @@ class GetParamGrad(nn.Cell):
super
(
GetParamGrad
,
self
).
__init__
(
auto_prefix
=
False
)
super
(
GetParamGrad
,
self
).
__init__
(
auto_prefix
=
False
)
self
.
network
=
network
self
.
network
=
network
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
weights
=
ParameterTuple
(
network
.
trainable_params
())
self
.
grad
=
C
.
GradOperation
(
'grad'
,
self
.
grad
=
C
.
GradOperation
(
get_by_list
=
True
,
get_by_list
=
True
,
sens_param
=
True
)
sens_param
=
True
)
def
construct
(
self
,
data
,
sens
):
def
construct
(
self
,
data
,
sens
):
...
...
tests/ut/python/nn/optim/test_lr_schedule.py
浏览文件 @
1a54785f
...
@@ -22,7 +22,7 @@ from mindspore.ops.operations import BiasAdd, MatMul
...
@@ -22,7 +22,7 @@ from mindspore.ops.operations import BiasAdd, MatMul
import
mindspore.ops.composite
as
C
import
mindspore.ops.composite
as
C
grad_by_list
=
C
.
GradOperation
(
'get_by_list'
,
get_by_list
=
True
)
grad_by_list
=
C
.
GradOperation
(
get_by_list
=
True
)
class
Net
(
Cell
):
class
Net
(
Cell
):
...
...
tests/ut/python/nn/test_nn_pad.py
浏览文件 @
1a54785f
...
@@ -34,7 +34,7 @@ class Net(nn.Cell):
...
@@ -34,7 +34,7 @@ class Net(nn.Cell):
class
Grad
(
nn
.
Cell
):
class
Grad
(
nn
.
Cell
):
def
__init__
(
self
,
network
):
def
__init__
(
self
,
network
):
super
(
Grad
,
self
).
__init__
()
super
(
Grad
,
self
).
__init__
()
self
.
grad
=
GradOperation
(
name
=
"get_all"
,
get_all
=
True
,
sens_param
=
True
)
self
.
grad
=
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
self
.
network
=
network
self
.
network
=
network
@
ms_function
@
ms_function
...
...
tests/ut/python/ops/test_bprop_disorder.py
浏览文件 @
1a54785f
...
@@ -28,7 +28,7 @@ from ....mindspore_test_framework.pipeline.forward.compile_forward \
...
@@ -28,7 +28,7 @@ from ....mindspore_test_framework.pipeline.forward.compile_forward \
import
pipeline_for_compile_forward_ge_graph_for_case_by_case_config
import
pipeline_for_compile_forward_ge_graph_for_case_by_case_config
grad_by_list_with_sens
=
C
.
GradOperation
(
'grad_by_list_with_sens'
,
get_by_list
=
True
,
sens_param
=
True
)
grad_by_list_with_sens
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
True
)
class
DisOrderTest1
(
nn
.
Cell
):
class
DisOrderTest1
(
nn
.
Cell
):
...
...
tests/ut/python/ops/test_control_ops.py
浏览文件 @
1a54785f
...
@@ -30,9 +30,9 @@ from mindspore.common import ms_function
...
@@ -30,9 +30,9 @@ from mindspore.common import ms_function
context
.
set_context
(
mode
=
context
.
GRAPH_MODE
)
context
.
set_context
(
mode
=
context
.
GRAPH_MODE
)
grad_by_list
=
C
.
GradOperation
(
'get_by_list'
,
get_by_list
=
True
)
grad_by_list
=
C
.
GradOperation
(
get_by_list
=
True
)
grad_all
=
C
.
GradOperation
(
'get_all'
,
get_all
=
True
)
grad_all
=
C
.
GradOperation
(
get_all
=
True
)
grad_all_with_sens
=
C
.
GradOperation
(
'grad_all_with_sens'
,
get_all
=
True
,
sens_param
=
True
)
grad_all_with_sens
=
C
.
GradOperation
(
get_all
=
True
,
sens_param
=
True
)
def
cond_data_test
(
x_init
,
y_init
):
def
cond_data_test
(
x_init
,
y_init
):
...
@@ -564,7 +564,7 @@ def test_switch_layer_env_eliminate():
...
@@ -564,7 +564,7 @@ def test_switch_layer_env_eliminate():
class
NetGrad
(
nn
.
Cell
):
class
NetGrad
(
nn
.
Cell
):
def
__init__
(
self
,
net
):
def
__init__
(
self
,
net
):
super
(
NetGrad
,
self
).
__init__
()
super
(
NetGrad
,
self
).
__init__
()
self
.
grad_op
=
C
.
GradOperation
(
'grad'
,
get_by_list
=
True
,
sens_param
=
False
)
self
.
grad_op
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
False
)
self
.
net
=
net
self
.
net
=
net
self
.
weights
=
ParameterTuple
(
self
.
net
.
trainable_params
())
self
.
weights
=
ParameterTuple
(
self
.
net
.
trainable_params
())
...
@@ -593,7 +593,7 @@ def test_switch_layer_single_layer():
...
@@ -593,7 +593,7 @@ def test_switch_layer_single_layer():
class
NetGrad
(
nn
.
Cell
):
class
NetGrad
(
nn
.
Cell
):
def
__init__
(
self
,
net
):
def
__init__
(
self
,
net
):
super
(
NetGrad
,
self
).
__init__
()
super
(
NetGrad
,
self
).
__init__
()
self
.
grad_op
=
C
.
GradOperation
(
'grad'
,
get_by_list
=
True
,
sens_param
=
False
)
self
.
grad_op
=
C
.
GradOperation
(
get_by_list
=
True
,
sens_param
=
False
)
self
.
net
=
net
self
.
net
=
net
self
.
weights
=
ParameterTuple
(
self
.
net
.
trainable_params
())
self
.
weights
=
ParameterTuple
(
self
.
net
.
trainable_params
())
...
...
tests/ut/python/ops/test_math_ops.py
浏览文件 @
1a54785f
...
@@ -38,7 +38,7 @@ context.set_context(mode=context.GRAPH_MODE)
...
@@ -38,7 +38,7 @@ context.set_context(mode=context.GRAPH_MODE)
# W0613: unused-argument
# W0613: unused-argument
# W0231: super-init-not-called
# W0231: super-init-not-called
grad
=
C
.
GradOperation
(
'grad'
)
grad
=
C
.
GradOperation
()
def
test_multiply
():
def
test_multiply
():
""" test_multiply """
""" test_multiply """
...
...
tests/ut/python/ops/test_momentum.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/ops/test_nn_ops.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/ops/test_ops.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_add_relu_redistribution.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_arithmetic.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_attention.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_auto_parallel_BN_PReLU.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_auto_parallel_arithmetic.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_auto_parallel_assign_sub_with_ref_key.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_auto_parallel_cast.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_auto_parallel_common_parameter.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_auto_parallel_double_sources.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_auto_parallel_double_star.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_auto_parallel_double_subgraphs.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_auto_parallel_fc_nobias.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_auto_parallel_four_matmul.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_auto_parallel_l2normalize.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_auto_parallel_matmul_drop.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_auto_parallel_matmul_prelu.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_auto_parallel_onehot.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_auto_parallel_partial_strategy.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_auto_parallel_reduce_method.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_auto_parallel_reshape.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_auto_parallel_rhombus.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_auto_parallel_softmax_loss.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_auto_parallel_transformer.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_auto_parallel_transpose.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_auto_parallel_tuple_depend.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_auto_parallel_two_matmul.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_auto_parallel_two_partial_matmul.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_auto_parallel_zig_zag.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_auto_star_elimination.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_batch_parallel.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_batch_parallel_dropout.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_batch_parallel_tensoradd.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_comparison_function_info.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_dataset_interface.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_different_type_for_div_op.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_element_wise_function.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_embeddinglookup.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_gather_v2.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_gather_v2_primitive.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_get_next.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_gpu_dropout.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_hybird_parallel_activation.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_l2normalize.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_linear.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_loop_two_matmul.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_loss_scale.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_matmul_dropout.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_matmul_tensor.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_mix_precision_hybrid_parallel.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_one_hot_net.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_one_weight_parameter.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_onehot.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_prelu.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_reduce_method_info.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_reshape.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_reshape_parameter.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_scalar_loss.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_semi_auto_two_subgraphs.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_softmax_cross_entropy_loss.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_sparse_feature_bprop.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_sparse_gather_v2.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_split_grad_sens.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_step_parallel.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_strategy_checkpoint.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_sum_as_loss.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_two_matmul.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_two_weights_parameter.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parallel/test_virtual_dataset_3_input.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parameter_feature/test_parameter.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/parameter_feature/test_var_grad.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/pipeline/infer/test_scalar_add_grad.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/pipeline/parse/test_parse.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/pynative_mode/ops/test_grad.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/pynative_mode/test_cont_cases.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/pynative_mode/test_framstruct.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/pynative_mode/test_high_order_grad.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/pynative_mode/test_hook.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/pynative_mode/test_implicit_conversion.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/pynative_mode/test_insert_grad_of.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/pynative_mode/test_kw_and_kwarg.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/pynative_mode/test_pynative_model.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/pynative_mode/test_sparse_pynative.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/pynative_mode/test_stop_gradient.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
tests/ut/python/pynative_mode/test_user_define_bprop_check.py
浏览文件 @
1a54785f
此差异已折叠。
点击以展开。
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录