Searched refs:rank_base (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ops/_utils/ |
| D | utils.py | 84 rank_base = len(x_shp[0])
85 validator.check_int_range(axis, -rank_base - 1, rank_base, Rel.INC_BOTH, 'axis', prim_name)
87 axis = axis + rank_base
94 for j in range(rank_base):
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | pack.cc | 27 int64_t rank_base = SizeToLong(output_shape.size()); in _get_pack_shape() local
30 axis = axis + rank_base + 1; in _get_pack_shape()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_arrays.cc | 106 int64_t rank_base = SizeToLong(shape->shape().size()); in InferImplStack() local
110 int64_t axis_value = CheckAxis(op_name, axis, -(rank_base + 1), rank_base); in InferImplStack()
112 axis_value = GetPositiveAxis(axis_value, LongToSize(rank_base + 1)); in InferImplStack()
950 int64_t rank_base = SizeToLong(shape_base.size()); in InferImplConcat() local
960 int64_t axis_value = CheckAxis(op_name, axis, -(rank_base + 1), rank_base); in InferImplConcat()
962 axis_value = GetPositiveAxis(axis_value, LongToSize(rank_base)); in InferImplConcat()
980 if (rank_tensor != rank_base) { in InferImplConcat()
983 for (int j = 0; j < rank_base; ++j) { in InferImplConcat()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | array_ops.py | 2530 rank_base = len(x_shape[0])
2533 validator.check_int_range(axis, -rank_base - 1, rank_base, Rel.INC_BOTH, 'axis', prim_name)
2535 axis = axis + rank_base + 1
|