Home
last modified time | relevance | path

Searched refs:axis2 (Results 1 – 25 of 28) sorted by relevance

12

/third_party/mindspore/tests/st/ops/cpu/
Dtest_reduce_op.py32 self.axis2 = -1
46 self.reduce_mean(indice, self.axis2),
50 self.reduce_sum(indice, self.axis2),
52 self.reduce_max(indice, self.axis2),
57 self.reduce_min(indice, self.axis2),
69 self.axis2 = (0, 1, 2)
78 self.reduce_all(indice, self.axis2),
82 self.reduce_any(indice, self.axis2),
Dtest_cumsum_op.py29 axis2 = 2 variable
43 list2 = [axis0, axis1, axis2, axis3, axis4, axis5, axis6]
55 self.axis2 = axis2
70 self.axis9 = axis2
85 self.axis16 = axis2
100 self.axis23 = axis2
115 self.axis30 = axis2
131 self.cumsum_op(self.x2, self.axis2),
Dtest_argmaxwithvalue_op.py29 axis2 = -1
31 self.argmax2 = P.ArgMaxWithValue(axis2)
/third_party/mindspore/mindspore/_extends/parse/
Dstandard_method.py356 def swapaxes(x, axis1, axis2): argument
382 axis1, axis2 = check_swapaxes_axis_const((axis1, axis2), x.ndim)
384 if axis1 == axis2:
386 if axis1 > axis2:
387 axis1, axis2 = axis2, axis1
391 if axis2 + 1 < x.ndim:
392 new_perm = perm[0:axis1] + perm[axis2:axis2 + 1] + \
393 perm[axis1 + 1:axis2] + perm[axis1:axis1 + 1] + perm[axis2 + 1:]
395 new_perm = perm[0:axis1] + perm[axis2:axis2 + 1] + \
396 perm[axis1 + 1:axis2] + perm[axis1:axis1 + 1]
[all …]
/third_party/mindspore/tests/st/ops/gpu/
Dtest_reduce_all_op.py35 axis2 = 1 variable
56 self.axis2 = axis2
68 P.ReduceAll(self.keep_dims2)(self.x2, self.axis2),
88 expect2 = np.all(x2, axis=axis2, keepdims=keep_dims2)
Dtest_reduce_any_op.py35 axis2 = 1 variable
56 self.axis2 = axis2
68 P.ReduceAny(self.keep_dims2)(self.x2, self.axis2),
88 expect2 = np.any(x2, axis=axis2, keepdims=keep_dims2)
Dtest_cumsum_op.py34 axis2 = 2
59 self.axis2 = axis2
77 P.CumSum()(self.x2, self.axis2),
99 expect2 = np.cumsum(x2, axis=axis2)
Dtest_cumprod_op.py34 axis2 = 2
59 self.axis2 = axis2
77 P.CumProd()(self.x2, self.axis2),
99 expect2 = np.cumprod(x2, axis=axis2)
Dtest_reduce_prod_op.py34 axis2 = 2 variable
78 self.axis2 = axis2
109 P.ReduceProd(self.keep_dims2)(self.x2, self.axis2),
132 expect2 = np.prod(x2, axis=axis2, keepdims=keep_dims2)
Dtest_reduce_min_op.py36 axis2 = 2 variable
80 self.axis2 = axis2
111 P.ReduceMin(self.keep_dims2)(self.x2, self.axis2),
140 expect2 = np.min(x2, axis=axis2, keepdims=keep_dims2)
Dtest_reduce_max_op.py36 axis2 = 2 variable
80 self.axis2 = axis2
111 P.ReduceMax(self.keep_dims2)(self.x2, self.axis2),
140 expect2 = np.max(x2, axis=axis2, keepdims=keep_dims2)
Dtest_reduce_sum_op.py35 axis2 = 2 variable
103 self.axis2 = axis2
158 P.ReduceSum(self.keep_dims2)(self.x2, self.axis2),
193 expect2 = np.sum(x2, axis=axis2, keepdims=keep_dims2)
Dtest_reduce_mean_op.py35 axis2 = 2 variable
101 self.axis2 = axis2
156 P.ReduceMean(self.keep_dims2)(self.x2, self.axis2),
191 expect2 = np.mean(x2, axis=axis2, keepdims=keep_dims2)
Dtest_argmaxwithvalue_op.py29 axis2 = -1
31 self.argmax2 = P.ArgMaxWithValue(axis2)
Dtest_argminwithvalue_op.py29 axis2 = -1
31 self.argmin2 = P.ArgMinWithValue(axis2)
Dtest_index_add_op.py75 axis2 = 2
79 net = NetIndexAdd(x, axis2)
83 net = NetIndexAdd(x, axis2)
/third_party/mindspore/mindspore/common/
Dtensor.py702 def swapaxes(self, axis1, axis2): argument
729 axis1, axis2 = validator.check_swapaxes_axis((axis1, axis2), self.ndim)
731 if axis1 == axis2:
733 if axis1 > axis2:
734 axis1, axis2 = axis2, axis1
737 if axis2 + 1 < self.ndim:
738 new_perm = perm[0:axis1] + perm[axis2:axis2+1] + \
739 perm[axis1+1:axis2] + perm[axis1:axis1+1] + perm[axis2+1:]
741 new_perm = perm[0:axis1] + perm[axis2:axis2+1] + \
742 perm[axis1+1:axis2] + perm[axis1:axis1+1]
[all …]
/third_party/mindspore/tests/st/ops/graph_kernel/
Dtest_reduce_max.py56 axis2 = 2
58 expect = get_output(x2, axis2, keep_dims2, False)
59 output = get_output(x2, axis2, keep_dims2, True)
Dtest_reduce_min.py56 axis2 = 2
58 expect = get_output(x2, axis2, keep_dims2, False)
59 output = get_output(x2, axis2, keep_dims2, True)
/third_party/mindspore/mindspore/numpy/
Darray_ops.py224 def swapaxes(x, axis1, axis2): argument
251 return x.swapaxes(axis1, axis2)
1988 axis1, axis2 = axes[0], axes[1]
1990 axis2 = _canonicalize_axis(axis2, a.ndim)
1991 if axis1 == axis2:
1997 return flip(flip(a, axis1), axis2)
1999 perm[axis1], perm[axis2] = perm[axis2], perm[axis1]
2002 return flip(transpose(a, perm), axis2)
Darray_creations.py1090 def diagonal(a, offset=0, axis1=0, axis2=1): argument
1146 return a.diagonal(offset=offset, axis1=axis1, axis2=axis2)
1149 def trace(a, offset=0, axis1=0, axis2=1, dtype=None): argument
1202 return a.trace(offset=offset, axis1=axis1, axis2=axis2, dtype=dtype)
/third_party/mindspore/tests/ut/cpp/parallel/ops_info/
Dsoftmax_info_test.cc60 ValuePtr axis2 = MakeValue(static_cast<int64_t>(4)); in SetUp() local
61 std::unordered_map<std::string, ValuePtr> attr2 = {{"axis", axis2}}; in SetUp()
/third_party/mindspore/tests/st/numpy_native/
Dtest_array_creations.py559 return mnp.diagonal(arr, offset=2, axis1=-1, axis2=0)
563 return onp.diagonal(arr, offset=2, axis1=-1, axis2=0)
576 match_res(mnp.diagonal, onp.diagonal, arr, offset=i, axis1=0, axis2=1)
577 match_res(mnp.diagonal, onp.diagonal, arr, offset=i, axis1=1, axis2=0)
581 match_res(mnp.diagonal, onp.diagonal, arr, offset=i, axis1=0, axis2=-1)
582 match_res(mnp.diagonal, onp.diagonal, arr, offset=i, axis1=-2, axis2=2)
584 offset=i, axis1=-1, axis2=-2)
588 return mnp.trace(arr, offset=4, axis1=1, axis2=2)
592 return onp.trace(arr, offset=4, axis1=1, axis2=2)
604 match_res(mnp.trace, onp.trace, arr, offset=-1, axis1=0, axis2=1)
[all …]
/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/cl/
Dconcat.cl240 CONCAT6(6input, axis2, _NHWC4)
241 CONCAT5(5input, axis2, _NHWC4)
242 CONCAT4(4input, axis2, _NHWC4)
243 CONCAT3(3input, axis2, _NHWC4)
244 CONCAT2(2input, axis2, _NHWC4)
/third_party/flutter/skia/third_party/externals/harfbuzz/test/shaping/data/text-rendering-tests/fonts/
DSelawik-README.md43 \axis2 | Shows 0 because this font doesn't yet have a second axis.
45 \axis2hex | Same as \axis2 but in hex.

12