Searched defs:axis2 (Results 1 – 13 of 13) sorted by relevance
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_cumsum_op.py | 29 axis2 = 2 variable
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_reduce_all_op.py | 35 axis2 = 1 variable
|
| D | test_reduce_any_op.py | 35 axis2 = 1 variable
|
| D | test_reduce_prod_op.py | 34 axis2 = 2 variable
|
| D | test_reduce_min_op.py | 36 axis2 = 2 variable
|
| D | test_reduce_max_op.py | 36 axis2 = 2 variable
|
| D | test_reduce_mean_op.py | 35 axis2 = 2 variable
|
| D | test_reduce_sum_op.py | 35 axis2 = 2 variable
|
| /third_party/mindspore/mindspore/_extends/parse/ |
| D | standard_method.py | 356 def swapaxes(x, axis1, axis2): argument
723 def diagonal(x, offset=0, axis1=0, axis2=1): argument
805 def trace(x, offset=0, axis1=0, axis2=1, dtype=None): argument
|
| /third_party/mindspore/mindspore/common/ |
| D | tensor.py | 702 def swapaxes(self, axis1, axis2): argument
1349 def diagonal(self, offset=0, axis1=0, axis2=1): argument
1431 def trace(self, offset=0, axis1=0, axis2=1, dtype=None): argument
|
| /third_party/mindspore/tests/ut/cpp/parallel/ops_info/ |
| D | softmax_info_test.cc | 60 ValuePtr axis2 = MakeValue(static_cast<int64_t>(4)); in SetUp() local
|
| /third_party/mindspore/mindspore/numpy/ |
| D | array_creations.py | 1090 def diagonal(a, offset=0, axis1=0, axis2=1): argument
1149 def trace(a, offset=0, axis1=0, axis2=1, dtype=None): argument
|
| D | array_ops.py | 224 def swapaxes(x, axis1, axis2): argument
|