| /third_party/mindspore/mindspore-src/source/mindspore/core/utils/ |
| D | shape_utils.h | 27 inline std::string ShapeVectorToString(const ShapeVector &shape) { in ShapeVectorToString()
36 inline size_t SizeOf(const ShapeVector &shape) { in SizeOf()
52 inline bool IsOneElementShape(const ShapeVector &shape) { in IsOneElementShape()
72 inline bool IsDynamicRank(const ShapeVector &shape) { in IsDynamicRank()
93 inline bool IsDynamicShape(const ShapeVector &shape) { in IsDynamicShape()
98 inline bool IsDynamic(const ShapeVector &shape) { in IsDynamic()
115 inline bool IsShapeEmpty(const ShapeVector &shape) { in IsShapeEmpty()
121 inline bool IsShapeNone(const ShapeVector &shape) { in IsShapeNone()
|
| /third_party/mindspore/mindspore-src/source/tests/ut/data/dataset/testTFTestAllTypes/ |
| D | datasetSchemaNoRow.json | 7 "shape": [1] array
12 "shape": [1] array
17 "shape": [1] array
22 "shape": [1] array
27 "shape": [2] array
32 "shape": [2, 2] array
37 "shape": [2, 2, 2] array
42 "shape": [1] array
|
| D | datasetSchema5Rows.json | 8 "shape": [1] array
13 "shape": [1] array
18 "shape": [1] array
23 "shape": [1] array
28 "shape": [2] array
33 "shape": [2, 2] array
38 "shape": [2, 2, 2] array
43 "shape": [1] array
|
| D | datasetSchema.json | 8 "shape": [1] array
13 "shape": [1] array
18 "shape": [1] array
23 "shape": [1] array
28 "shape": [2] array
33 "shape": [2, 2] array
38 "shape": [2, 2, 2] array
43 "shape": [1] array
|
| D | datasetSchema7Rows.json | 8 "shape": [1] array
13 "shape": [1] array
18 "shape": [1] array
23 "shape": [1] array
28 "shape": [2] array
33 "shape": [2, 2] array
38 "shape": [2, 2, 2] array
43 "shape": [1] array
|
| D | datasetSchema1Row.json | 8 "shape": [1] array
13 "shape": [1] array
18 "shape": [1] array
23 "shape": [1] array
28 "shape": [2] array
33 "shape": [2, 2] array
38 "shape": [2, 2, 2] array
43 "shape": [1] array
|
| /third_party/mindspore/mindspore-src/source/tests/ut/data/dataset/testBatchDataset/ |
| D | datasetSchema.json | 8 "shape": [1] array
13 "shape": [1] array
18 "shape": [1] array
23 "shape": [1] array
28 "shape": [2] array
33 "shape": [2, 2] array
38 "shape": [2, 2, 2] array
43 "shape": [1] array
|
| /third_party/mindspore/mindspore-src/source/tests/ut/data/dataset/testTFBert5Rows/ |
| D | datasetSchema.json | 8 "shape": [1] array
13 "shape": [20] array
18 "shape": [20] array
23 "shape": [20] array
28 "shape": [128] array
33 "shape": [128] array
38 "shape": [128] array
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/ |
| D | test_inplace_op.py | 71 def test_inplace_add(shape, indice_len, dtype): argument
93 def test_inplace_add_same_indice(shape, dtype): argument
115 def test_inplace_add_1d(shape, indice, dtype): argument
136 def test_inplace_sub(shape, indice_len, dtype): argument
158 def test_inplace_sub_same_indice(shape, dtype): argument
180 def test_inplace_sub_1d(shape, indice, dtype): argument
200 def test_inplace_update(shape, indice_len, dtype): argument
222 def test_vmap_inplace_update(shape, indice_len, dtype): argument
247 def test_vmap_inplace_add(shape, indice_len, dtype): argument
272 def test_vmap_inplace_sub(shape, indice_len, dtype): argument
|
| D | test_bitwise_op.py | 51 def test_bitwise_and(shape, dtype, mode_cpu): argument
78 def test_bitwise_or(shape, dtype, mode_cpu): argument
105 def test_bitwise_xor(shape, dtype, mode_cpu): argument
165 def test_bitwise_and_tensor_interface_operator(dtype, mode, shape): argument
187 def test_bitwise_or_tensor_interface_operator(dtype, mode, shape): argument
209 def test_bitwise_xor_tensor_interface_operator(dtype, mode, shape): argument
231 def test_bitwise_and_tensor_interface(dtype, mode, shape): argument
253 def test_bitwise_or_tensor_interface(dtype, mode, shape): argument
275 def test_bitwise_xor_tensor_interface(dtype, mode, shape): argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/data/dataset/testTFBert5Rows1/ |
| D | datasetSchema.json | 8 "shape": [1] array
13 "shape": [20] array
18 "shape": [20] array
23 "shape": [20] array
28 "shape": [128] array
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/parallel/ |
| D | device_matrix_test.cc | 35 Shape shape = {2, 3}; in TEST_F() local
46 Shape shape = {2, 2, 3}; in TEST_F() local
57 Shape shape = {2, 1, 4, 2}; in TEST_F() local
69 Shape shape = {3, 4, 2, 3, 2}; in TEST_F() local
81 Shape shape = {2, 2, 2}; in TEST_F() local
90 Shape shape = {4, 2}; in TEST_F() local
101 Shape shape = {4, 2}; in TEST_F() local
112 Shape shape = {4, 2}; in TEST_F() local
122 Shape shape = {2, 2, 2}; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/mindapi/src/ |
| D | tensor.cc | 27 Tensor::Tensor(TypeId data_type, const ShapeVector &shape) : Value(std::make_shared<TensorImpl>(dat… in Tensor()
29 Tensor::Tensor(TypeId data_type, const ShapeVector &shape, void *data, size_t data_len) in Tensor()
32 const ShapeVector &Tensor::shape() const { return ToRef<TensorImpl>(impl_).shape(); } in shape() function in mindspore::api::Tensor
34 void Tensor::set_shape(const ShapeVector &shape) { (void)ToRef<TensorImpl>(impl_).set_shape(shape);… in set_shape()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/ |
| D | shape.c | 21 ShapeStruct *shape = (ShapeStruct *)self; in ShapeCompute() local
29 ShapeStruct *shape = (ShapeStruct *)self; in ShapeResize() local
36 ShapeStruct *shape = (ShapeStruct *)malloc(sizeof(ShapeStruct)); in CreateShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/utils/ |
| D | op_util.cc | 43 std::vector<int64_t> ToVector(const ge::Shape &shape) { in ToVector()
53 std::string ToString(const ge::Shape &shape) { return ge::DebugString(ToVector(shape)); } in ToString()
55 std::string ToString(const ge::Shape *shape) { return ge::DebugString(ToVector(*shape)); } in ToString()
57 std::string ToString(const std::vector<int64_t> &shape) { return ge::DebugString(shape); } in ToString()
61 for (ge::Shape shape : shapes) { in ToString() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/ |
| D | common_func.c | 19 int Offset(const int *shape, const int dim0, const int dim1, const int dim2, const int dim3) { in Offset()
23 int64_t OffsetComm(const int *shape, const int dim0, const int dim1, const int dim2) { in OffsetComm()
27 int Offset4d(const int *shape, const int *dims) { return Offset(shape, dims[0], dims[1], dims[2], d… in Offset4d()
29 int64_t Offset6d(const int *shape, const int *dims) { in Offset6d()
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ir/ |
| D | meta_tensor.h | 119 const ShapeVector &shape() const { return shape_; } in shape() function
129 virtual size_t set_shape(const ShapeVector &shape) { in set_shape()
264 const ShapeVector &shape() const { return shape_; } in shape() function
269 void set_shape(const ShapeVector &shape) { this->shape_ = shape; } in set_shape()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/base/ |
| D | group_convolution_creator.h | 46 auto shape = origin_outputs_.front()->shape(); in GroupConvCreator() local
60 void set_input_shape(const std::vector<int> &shape) { input_shape_ = shape; } in set_input_shape()
61 void set_output_shape(const std::vector<int> &shape) { output_shape_ = shape; } in set_output_shape()
62 void set_filter_shape(const std::vector<int> &shape) { filter_shape_ = shape; } in set_filter_shape()
63 void set_bias_shape(const std::vector<int> &shape) { bias_shape_ = shape; } in set_bias_shape()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/common/ |
| D | _utils.py | 29 def is_shape_unknown(shape): argument
40 def is_dim_unknown(shape): argument
49 def get_slice_num(dtype, shape): argument
74 def get_slice_shape(dtype, shape): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/sparse/ |
| D | test_coo.py | 82 def __init__(self, shape): argument
120 def __init__(self, shape): argument
439 def test_coo_neg(indices, values, shape): argument
443 def test_coo_add_coo(indices, indices_2, values, values_2, shape): argument
448 def test_coo_add_dense(indices, values, shape, dense): argument
452 def test_dense_add_coo(indices, values, shape, dense): argument
456 def test_coo_sub_coo(indices, indices_2, values, values_2, shape): argument
461 def test_coo_sub_dense(indices, values, shape, dense): argument
465 def test_dense_sub_coo(indices, values, shape, dense): argument
469 def test_coo_mul_dense(indices, values, shape, dense): argument
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/ |
| D | gpu_kernel.cc | 159 void ShapeNCHW2NHWC(ShapeVector *shape) { in ShapeNCHW2NHWC()
164 void ShapeNCDHW2NDHWC(ShapeVector *shape) { in ShapeNCDHW2NDHWC()
171 void SetDimA(const ShapeVector &shape, int *dimA, size_t len, const std::string &format) { in SetDimA()
189 void SetStrideA(const ShapeVector &shape, int *strideA, size_t len, const std::string &format) { in SetStrideA()
207 void SetNCHW(const ShapeVector &shape, int *n, int *c, int *h, int *w, const std::string &format) { in SetNCHW()
223 void SetNCDHW(const ShapeVector &shape, int *n, int *c, int *d, int *h, int *w, const std::string &… in SetNCDHW()
242 void SetDimA(const ShapeVector &shape, int *dimA, size_t len, const mindspore::Format &format) { in SetDimA()
260 void SetStrideA(const ShapeVector &shape, int *strideA, size_t len, const mindspore::Format &format… in SetStrideA()
278 void SetNCHW(const ShapeVector &shape, int *n, int *c, int *h, int *w, const mindspore::Format &for… in SetNCHW()
294 void SetNCDHW(const ShapeVector &shape, int *n, int *c, int *d, int *h, int *w, const mindspore::Fo… in SetNCDHW()
[all …]
|
| /third_party/mindspore/mindspore-src/source/tests/ut/data/dataset/testDataset2/ |
| D | datasetSchema.json | 8 "shape": [3, 4, 2], array
14 "shape": [7], array
20 "shape": [1, 13, 14, 12], array
26 "shape": [9], array
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fisson/ |
| D | node_out_shapes.cc | 45 const auto &shape = in_shape->cast<abstract::ShapePtr>()->shape(); in Run() local
58 auto shape = out_shape->cast<abstract::TupleShapePtr>(); in Run() local
69 const auto &shape = out_shape->cast<abstract::ShapePtr>()->shape(); in Run() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/common/expander/core/ |
| D | node.cc | 37 std::vector<int64_t> Node::shape() { in shape() function in mindspore::expander::Node
45 auto shape = shape_->cast<abstract::ShapePtr>(); in shape() local
63 auto shape = bs->cast<abstract::ShapePtr>(); in shapes() local
128 auto shape = emitter()->infer()->GetShape(shared_from_this()); in GetShape() local
165 auto shape = abstract()->BuildShape(); in GetShape() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/dyn_shape_dev/ |
| D | test_broadcast_to.py | 26 def broadcast_to_forward_func(x, shape): argument
31 def broadcast_to_backward_func(x, shape): argument
36 def broadcast_to_dyn_shape_func(x, shape): argument
|