Searched refs:reduced_shape (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_math_ops.py | 26 from ..functional import broadcast_gradient_args, reduced_shape, tuple_div
67 output_shape_kept_dims = reduced_shape(input_shape, axis)
76 output_shape_kept_dims = reduced_shape(input_shape, axis)
669 output_shape_kept_dims = reduced_shape(input_shape, axis)
|
| /third_party/mindspore/mindspore/nn/loss/ |
| D | loss.py | 1062 def _check_reduced_shape_valid(ori_shape, reduced_shape, axis, cls_name): argument
1064 validator.check_reduce_shape(ori_shape, reduced_shape, axis, cls_name)
|
| /third_party/mindspore/mindspore/ops/ |
| D | functional.py | 240 reduced_shape = Primitive("reduced_shape") variable
|