Home
last modified time | relevance | path

Searched defs:GradWrap (Results 1 – 25 of 114) sorted by relevance

12345

/third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/
Dtest_strategy_checkpoint.py42 class GradWrap(nn.Cell): class
105 class GradWrap(nn.Cell): class
174 class GradWrap(nn.Cell): class
239 class GradWrap(nn.Cell): class
306 class GradWrap(nn.Cell): class
Dtest_auto_parallel_assign_sub_with_ref_key.py46 class GradWrap(nn.Cell): class
Dtest_scalar_loss.py34 class GradWrap(nn.Cell): class
Dtest_auto_parallel_BN_PReLU.py49 class GradWrap(nn.Cell): class
Dtest_auto_parallel_segment_sum.py45 class GradWrap(nn.Cell): class
Dtest_auto_parallel_segment_min.py45 class GradWrap(nn.Cell): class
Dtest_auto_parallel_two_partial_matmul.py45 class GradWrap(nn.Cell): class
Dtest_auto_parallel_softmax_loss.py44 class GradWrap(nn.Cell): class
Dtest_batch_parallel_tensoradd.py45 class GradWrap(nn.Cell): class
Dtest_auto_parallel_matmul_drop.py45 class GradWrap(nn.Cell): class
Dtest_batch_parallel_dropout.py45 class GradWrap(nn.Cell): class
Dtest_auto_parallel_common_parameter.py45 class GradWrap(nn.Cell): class
Dtest_auto_parallel_fc_nobias.py45 class GradWrap(nn.Cell): class
Dtest_auto_parallel_matmul_prelu.py47 class GradWrap(nn.Cell): class
Dtest_dynamic_shape.py48 class GradWrap(nn.Cell): class
Dtest_auto_parallel_transpose.py46 class GradWrap(nn.Cell): class
Dtest_auto_parallel_l2normalize.py46 class GradWrap(nn.Cell): class
Dtest_auto_parallel_two_matmul.py51 class GradWrap(nn.Cell): class
/third_party/mindspore/mindspore-src/source/tests/ut/python/pynative_mode/
Dtest_sparse_pynative.py36 class GradWrap(nn.Cell): class
/third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/test_tbe_ops/
Dtest_minimum_grad.py39 class GradWrap(Cell): class
Dtest_maximum_grad.py39 class GradWrap(Cell): class
Dtest_select.py40 class GradWrap(Cell): class
/third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/
Dtest_dynamic_broadcast_gradient_args.py67 class GradWrap(nn.Cell): class
/third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/
Dtest_dynamic_broadcast_gradient_args.py69 class GradWrap(nn.Cell): class
/third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/
Dtest_minimum_grad_op.py39 class GradWrap(Cell): class

12345