Searched refs:reduce_max (Results 1 – 21 of 21) sorted by relevance
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_reduce_op.py | 39 self.reduce_max = P.ReduceMax(False)
51 self.reduce_max(indice, self.axis0),
52 self.reduce_max(indice, self.axis2),
53 self.reduce_max(indice, self.axis5),
54 self.reduce_max(indice, self.axis6),
|
| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_reduce_max.py | 27 self.reduce_max = P.ReduceMax(keep_dims)
30 return self.reduce_max(x, axis)
|
| D | test_optimize_assign.py | 30 self.reduce_max = P.ReduceMax()
38 return self.reduce_max(add_res)
|
| /third_party/mindspore/mindspore/ops/_op_impl/akg/ascend/ |
| D | __init__.py | 38 from .reduce_max import _reduce_max_akg
|
| /third_party/mindspore/mindspore/ops/_op_impl/cpu/ |
| D | __init__.py | 55 from .reduce_max import _reduce_max_cpu
|
| /third_party/mindspore/tests/st/networks/models/bert/src/ |
| D | CRF.py | 70 self.reduce_max = P.ReduceMax(keep_dims=True)
79 max_score = self.reduce_max(logits, -1) #16 5 5
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_reduce_prod_op.py | 123 reduce_max = ReduceProd()
124 output = reduce_max()
|
| D | test_reduce_max_op.py | 125 reduce_max = ReduceMax()
126 output = reduce_max()
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_one_hot_net.py | 141 self.reduce_max = P.ReduceMax(keep_dims=True)
142 self.reduce_max.shard(strategy=strategy.twod_strategy)
185 max_o = self.reduce_max(mul_const5_o, -1)
|
| D | test_reduce_method_info.py | 265 self.reduce_max = P.ReduceMax(keep_dims=False).shard(strategy2)
270 out = self.reduce_max(out, -1)
567 self.reduce_max = P.ReduceMax(keep_dims=False).shard(strategy2)
572 out = self.reduce_max(out)
|
| /third_party/mindspore/tests/st/auto_parallel/ |
| D | soft_entropy_loss_expand_parallel.py | 155 self.reduce_max = P.ReduceMax(keep_dims=True).shard(strategy=stra_list[9])
159 logit_max = self.reduce_max(logit, -1)
|
| /third_party/mindspore/tests/ut/python/dtype/ |
| D | test_list.py | 225 self.reduce_max = P.ReduceMax()
233 ret_max = self.reduce_max(x, self.axis)
|
| /third_party/mindspore/mindspore/ops/_op_impl/_custom_op/ |
| D | minmax_update_perchannel.py | 66 x_max = te.lang.cce.reduce_max(x, axis=axis)
|
| D | minmax_update_perlayer.py | 63 x_max = te.lang.cce.reduce_max(x, axis=axis)
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3_darknet53/src/ |
| D | yolo.py | 294 self.reduce_max = P.ReduceMax(keep_dims=False)
331 best_iou = self.reduce_max(iou, -1)
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3/src/ |
| D | yolov3.py | 518 self.reduce_max = P.ReduceMax(keep_dims=False)
542 best_iou = self.reduce_max(iou, -1) # [batch, grid[0], grid[1], num_anchor]
|
| /third_party/mindspore/mindspore/ops/ |
| D | functional.py | 92 reduce_max = P.ReduceMax() variable
|
| /third_party/mindspore/mindspore/ops/_op_impl/tbe/ |
| D | __init__.py | 155 from .reduce_max import _reduce_max_tbe
|
| /third_party/mindspore/mindspore/common/ |
| D | tensor.py | 1006 reduce_max = tensor_operator_registry.get("reduce_max")
1008 return reduce_(self, reduce_max(keepdims), cmp_fn=maximum(), axis=axis, keepdims=keepdims,
|
| /third_party/mindspore/mindspore/numpy/ |
| D | array_creations.py | 2110 end = F.reduce_max(a)
|
| D | math_ops.py | 4642 … length = int(maximum(F.reduce_max(x.astype(mstype.float32)), minlength - 1).asnumpy()) + 1
|