| /third_party/mindspore/mindspore/lite/test/ut/nnacl/infer/ |
| D | concat_infer_test.cc | 28 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
29 inputs[0] = new TensorC; in TEST_F()
30 inputs[0]->shape_size_ = 2; in TEST_F()
31 inputs[0]->shape_[0] = 3; in TEST_F()
32 inputs[0]->shape_[1] = 4; in TEST_F()
33 inputs[1] = new TensorC; in TEST_F()
34 inputs[1]->shape_size_ = 2; in TEST_F()
35 inputs[1]->shape_[0] = 3; in TEST_F()
36 inputs[1]->shape_[1] = 4; in TEST_F()
41 …int ret = ConcatInferShape((const TensorC **)inputs.data(), inputs.size(), outputs.data(), outputs… in TEST_F()
[all …]
|
| D | conv2d_infer_test.cc | 28 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
29 inputs[0] = new TensorC; in TEST_F()
30 inputs[0]->shape_size_ = 4; in TEST_F()
31 inputs[0]->shape_[0] = 5; in TEST_F()
32 inputs[0]->shape_[1] = 4; in TEST_F()
33 inputs[0]->shape_[2] = 4; in TEST_F()
34 inputs[0]->shape_[3] = 6; in TEST_F()
35 inputs[1] = new TensorC; in TEST_F()
36 inputs[1]->shape_size_ = 4; in TEST_F()
37 inputs[1]->shape_[0] = 20; in TEST_F()
[all …]
|
| D | depthwise_conv2d_infer_test.cc | 28 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
29 inputs[0] = new TensorC; in TEST_F()
30 inputs[0]->shape_size_ = 4; in TEST_F()
31 inputs[0]->shape_[0] = 5; in TEST_F()
32 inputs[0]->shape_[1] = 4; in TEST_F()
33 inputs[0]->shape_[2] = 4; in TEST_F()
34 inputs[0]->shape_[3] = 6; in TEST_F()
35 inputs[1] = new TensorC; in TEST_F()
36 inputs[1]->shape_size_ = 4; in TEST_F()
37 inputs[1]->shape_[0] = 6; // in channel in TEST_F()
[all …]
|
| D | arithmetic_infer_test.cc | 28 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
29 inputs[0] = new TensorC; in TEST_F()
30 inputs[0]->shape_size_ = 4; in TEST_F()
31 inputs[0]->shape_[0] = 2; in TEST_F()
32 inputs[0]->shape_[1] = 3; in TEST_F()
33 inputs[0]->shape_[2] = 4; in TEST_F()
34 inputs[0]->shape_[3] = 5; in TEST_F()
35 inputs[1] = new TensorC; in TEST_F()
36 inputs[1]->shape_size_ = 5; in TEST_F()
37 inputs[1]->shape_[0] = 6; in TEST_F()
[all …]
|
| D | arithmetic_compare_infer_test.cc | 28 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
29 inputs[0] = new TensorC; in TEST_F()
30 inputs[0]->shape_size_ = 4; in TEST_F()
31 inputs[0]->shape_[0] = 2; in TEST_F()
32 inputs[0]->shape_[1] = 3; in TEST_F()
33 inputs[0]->shape_[2] = 4; in TEST_F()
34 inputs[0]->shape_[3] = 5; in TEST_F()
35 inputs[1] = new TensorC; in TEST_F()
36 inputs[1]->shape_size_ = 5; in TEST_F()
37 inputs[1]->shape_[0] = 6; in TEST_F()
[all …]
|
| D | reshape_infer_test.cc | 29 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
30 inputs[0] = new TensorC; in TEST_F()
31 inputs[0]->shape_size_ = 2; in TEST_F()
32 inputs[0]->shape_[0] = 2; in TEST_F()
33 inputs[0]->shape_[1] = 3; in TEST_F()
39 …int ret = ReshapeInferShape((const TensorC **)inputs.data(), inputs.size(), outputs.data(), output… in TEST_F()
46 delete inputs[i]; in TEST_F()
55 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
56 inputs[0] = new TensorC; in TEST_F()
57 inputs[0]->shape_size_ = 2; in TEST_F()
[all …]
|
| D | gather_nd_infer_test.cc | 28 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
29 inputs[0] = new TensorC; in TEST_F()
30 inputs[0]->shape_size_ = 3; in TEST_F()
31 inputs[0]->shape_[0] = 3; in TEST_F()
32 inputs[0]->shape_[1] = 2; in TEST_F()
33 inputs[0]->shape_[2] = 3; in TEST_F()
34 inputs[1] = new TensorC; in TEST_F()
35 inputs[1]->shape_size_ = 2; in TEST_F()
36 inputs[1]->shape_[0] = 2; in TEST_F()
37 inputs[1]->shape_[1] = 2; in TEST_F()
[all …]
|
| D | resize_infer_test.cc | 28 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
29 inputs[0] = new TensorC; in TEST_F()
30 inputs[0]->shape_size_ = 4; in TEST_F()
31 inputs[0]->shape_[0] = 4; in TEST_F()
32 inputs[0]->shape_[1] = 5; in TEST_F()
33 inputs[0]->shape_[2] = 3; in TEST_F()
34 inputs[0]->shape_[3] = 5; in TEST_F()
35 inputs[0]->format_ = Format_NHWC; in TEST_F()
41 …int ret = ResizeInferShape((const TensorC **)inputs.data(), inputs.size(), outputs.data(), outputs… in TEST_F()
51 delete inputs[i]; in TEST_F()
[all …]
|
| D | matmul_infer_test.cc | 28 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
29 inputs[0] = new TensorC; in TEST_F()
30 inputs[0]->shape_size_ = 2; in TEST_F()
31 inputs[0]->shape_[0] = 4; in TEST_F()
32 inputs[0]->shape_[1] = 3; in TEST_F()
33 inputs[1] = new TensorC; in TEST_F()
34 inputs[1]->shape_size_ = 2; in TEST_F()
35 inputs[1]->shape_[0] = 4; in TEST_F()
36 inputs[1]->shape_[1] = 3; in TEST_F()
42 …int ret = MatmulInferShape((const TensorC **)inputs.data(), inputs.size(), outputs.data(), outputs… in TEST_F()
[all …]
|
| D | gather_infer_test.cc | 28 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
29 inputs[0] = new TensorC; in TEST_F()
30 inputs[0]->shape_size_ = 2; in TEST_F()
31 inputs[0]->shape_[0] = 18; in TEST_F()
32 inputs[0]->shape_[1] = 3; in TEST_F()
33 inputs[1] = new TensorC; in TEST_F()
34 inputs[1]->shape_size_ = 3; in TEST_F()
35 inputs[1]->shape_[0] = 2; in TEST_F()
36 inputs[1]->shape_[1] = 3; in TEST_F()
37 inputs[1]->shape_[2] = 2; in TEST_F()
[all …]
|
| D | deconv2d_infer_test.cc | 28 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
29 inputs[0] = new TensorC; in TEST_F()
30 inputs[0]->shape_size_ = 4; in TEST_F()
31 inputs[0]->shape_[0] = 5; in TEST_F()
32 inputs[0]->shape_[1] = 4; in TEST_F()
33 inputs[0]->shape_[2] = 4; in TEST_F()
34 inputs[0]->shape_[3] = 6; in TEST_F()
35 inputs[0]->format_ = Format_NHWC; in TEST_F()
36 inputs[1] = new TensorC; in TEST_F()
37 inputs[1]->shape_size_ = 4; in TEST_F()
[all …]
|
| D | strided_slice_infer_test.cc | 28 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
29 inputs[0] = new TensorC; in TEST_F()
30 inputs[0]->shape_size_ = 3; in TEST_F()
31 inputs[0]->shape_[0] = 3; in TEST_F()
32 inputs[0]->shape_[1] = 3; in TEST_F()
33 inputs[0]->shape_[2] = 3; in TEST_F()
52 …int ret = StridedSliceInferShape((const TensorC **)inputs.data(), inputs.size(), outputs.data(), o… in TEST_F()
61 delete inputs[i]; in TEST_F()
70 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
71 inputs[0] = new TensorC; in TEST_F()
[all …]
|
| D | pad_infer_test.cc | 28 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
29 inputs[0] = new TensorC; in TEST_F()
30 inputs[0]->shape_size_ = 2; in TEST_F()
31 inputs[0]->shape_[0] = 2; in TEST_F()
32 inputs[0]->shape_[1] = 3; in TEST_F()
33 inputs[1] = new TensorC; in TEST_F()
35 inputs[1]->data_ = padding_tensor.data(); in TEST_F()
36 inputs[1]->shape_size_ = 2; in TEST_F()
37 inputs[1]->shape_[0] = 1; in TEST_F()
38 inputs[1]->shape_[1] = 4; in TEST_F()
[all …]
|
| D | crop_and_resize_infer_test.cc | 34 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
35 inputs[0] = new TensorC; in TEST_F()
36 inputs[0]->shape_size_ = 4; in TEST_F()
37 inputs[0]->shape_[0] = 3; in TEST_F()
38 inputs[0]->shape_[1] = 4; in TEST_F()
39 inputs[0]->shape_[2] = 5; in TEST_F()
40 inputs[0]->shape_[3] = 6; in TEST_F()
41 inputs[0]->data_type_ = kNumberTypeInt32; in TEST_F()
42 inputs[0]->format_ = Format_NHWC; in TEST_F()
43 inputs[1] = new TensorC; in TEST_F()
[all …]
|
| D | gru_infer_test.cc | 28 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
29 inputs[0] = new TensorC; in TEST_F()
30 inputs[0]->shape_size_ = 3; in TEST_F()
31 inputs[0]->shape_[0] = 4; in TEST_F()
32 inputs[0]->shape_[1] = 5; in TEST_F()
33 inputs[0]->shape_[2] = 6; in TEST_F()
34 inputs[0]->data_type_ = kNumberTypeInt32; in TEST_F()
35 inputs[0]->format_ = Format_NHWC; in TEST_F()
36 inputs[1] = new TensorC; in TEST_F()
37 inputs[1]->shape_size_ = 3; in TEST_F()
[all …]
|
| D | depth_to_space_infer_test.cc | 29 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
30 inputs[0] = new TensorC; in TEST_F()
31 inputs[0]->format_ = Format_NHWC; in TEST_F()
32 inputs[0]->shape_size_ = 4; in TEST_F()
33 inputs[0]->shape_[0] = 1; in TEST_F()
34 inputs[0]->shape_[1] = 1; in TEST_F()
35 inputs[0]->shape_[2] = 1; in TEST_F()
36 inputs[0]->shape_[3] = 12; in TEST_F()
41 …int ret = DepthToSpaceInferShape((const TensorC **)inputs.data(), inputs.size(), outputs.data(), o… in TEST_F()
51 delete inputs[i]; in TEST_F()
[all …]
|
| D | range_infer_test.cc | 29 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
30 inputs[0] = new TensorC; in TEST_F()
31 inputs[0]->shape_size_ = 2; in TEST_F()
32 inputs[0]->shape_[0] = 4; in TEST_F()
33 inputs[0]->shape_[1] = 3; in TEST_F()
40 …int ret = RangeInferShape((const TensorC **)inputs.data(), inputs.size(), outputs.data(), outputs.… in TEST_F()
47 delete inputs[i]; in TEST_F()
56 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
60 inputs[0] = new TensorC; in TEST_F()
61 inputs[0]->shape_size_ = 1; in TEST_F()
[all …]
|
| D | squeeze_infer_test.cc | 28 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
29 inputs[0] = new TensorC; in TEST_F()
30 inputs[0]->shape_size_ = 5; in TEST_F()
31 inputs[0]->shape_[0] = 2; in TEST_F()
32 inputs[0]->shape_[1] = 1; in TEST_F()
33 inputs[0]->shape_[2] = 3; in TEST_F()
34 inputs[0]->shape_[3] = 1; in TEST_F()
35 inputs[0]->shape_[4] = 4; in TEST_F()
40 …int ret = SqueezeInferShape((const TensorC **)inputs.data(), inputs.size(), outputs.data(), output… in TEST_F()
49 delete inputs[i]; in TEST_F()
[all …]
|
| D | lsh_projection_infer_test.cc | 28 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
29 inputs[0] = new TensorC; in TEST_F()
30 inputs[0]->shape_size_ = 2; in TEST_F()
31 inputs[0]->shape_[0] = 4; in TEST_F()
32 inputs[0]->shape_[1] = 3; in TEST_F()
33 inputs[0]->data_type_ = kNumberTypeInt32; in TEST_F()
34 inputs[0]->format_ = Format_NHWC; in TEST_F()
35 inputs[1] = new TensorC; in TEST_F()
36 inputs[1]->shape_size_ = 2; in TEST_F()
41 …int ret = LshProjectionInferShape((const TensorC **)inputs.data(), inputs.size(), outputs.data(), … in TEST_F()
[all …]
|
| D | slice_infer_test.cc | 28 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
29 inputs[0] = new TensorC; in TEST_F()
30 inputs[0]->shape_size_ = 2; in TEST_F()
31 inputs[0]->shape_[0] = 4; in TEST_F()
32 inputs[0]->shape_[1] = 4; in TEST_F()
42 …int ret = SliceInferShape((const TensorC **)inputs.data(), inputs.size(), outputs.data(), outputs.… in TEST_F()
50 delete inputs[i]; in TEST_F()
59 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
60 inputs[0] = new TensorC; in TEST_F()
61 inputs[0]->shape_size_ = 3; in TEST_F()
[all …]
|
| D | full_connection_infer_test.cc | 29 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
30 inputs[0] = new TensorC; in TEST_F()
31 inputs[0]->shape_size_ = 4; in TEST_F()
32 inputs[0]->shape_[0] = 1; in TEST_F()
33 inputs[0]->shape_[1] = 64; in TEST_F()
34 inputs[0]->shape_[2] = 5; in TEST_F()
35 inputs[0]->shape_[3] = 5; in TEST_F()
36 inputs[1] = new TensorC; in TEST_F()
37 inputs[1]->shape_size_ = 2; in TEST_F()
38 inputs[1]->shape_[0] = 256; in TEST_F()
[all …]
|
| D | power_infer_test.cc | 28 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
29 inputs[0] = new TensorC; in TEST_F()
30 inputs[0]->shape_size_ = 2; in TEST_F()
31 inputs[0]->shape_[0] = 4; in TEST_F()
32 inputs[0]->shape_[1] = 3; in TEST_F()
36 …int ret = PowerInferShape((const TensorC **)inputs.data(), inputs.size(), outputs.data(), outputs.… in TEST_F()
44 delete inputs[i]; in TEST_F()
53 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
54 inputs[0] = new TensorC; in TEST_F()
55 inputs[0]->shape_size_ = 2; in TEST_F()
[all …]
|
| D | batch_to_space_infer_test.cc | 29 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
30 inputs[0] = new TensorC; in TEST_F()
31 inputs[0]->shape_size_ = 4; in TEST_F()
32 inputs[0]->shape_[0] = 4; in TEST_F()
33 inputs[0]->shape_[1] = 1; in TEST_F()
34 inputs[0]->shape_[2] = 1; in TEST_F()
35 inputs[0]->shape_[3] = 1; in TEST_F()
36 inputs[0]->data_type_ = kNumberTypeInt32; in TEST_F()
37 inputs[0]->format_ = Format_NHWC; in TEST_F()
47 …int ret = BatchToSpaceInferShape((const TensorC **)inputs.data(), inputs.size(), outputs.data(), o… in TEST_F()
[all …]
|
| D | mean_infer_test.cc | 29 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
30 inputs[0] = new TensorC; in TEST_F()
31 inputs[0]->shape_size_ = 2; in TEST_F()
32 inputs[0]->shape_[0] = 2; in TEST_F()
33 inputs[0]->shape_[1] = 3; in TEST_F()
40 …int ret = MeanInferShape((const TensorC **)inputs.data(), inputs.size(), outputs.data(), outputs.s… in TEST_F()
47 delete inputs[i]; in TEST_F()
56 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
57 inputs[0] = new TensorC; in TEST_F()
58 inputs[0]->shape_size_ = 2; in TEST_F()
[all …]
|
| D | unsqueeze_infer_test.cc | 29 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
30 inputs[0] = new TensorC; in TEST_F()
31 inputs[0]->shape_size_ = 1; in TEST_F()
32 inputs[0]->shape_[0] = 4; in TEST_F()
38 …int ret = UnsqueezeInferShape((const TensorC **)inputs.data(), inputs.size(), outputs.data(), outp… in TEST_F()
46 delete inputs[i]; in TEST_F()
55 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
56 inputs[0] = new TensorC; in TEST_F()
57 inputs[0]->shape_size_ = 1; in TEST_F()
58 inputs[0]->shape_[0] = 4; in TEST_F()
[all …]
|