| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | space_to_depth_tests.cc | 44 std::vector<int> output_shape = {1, 1, 1, 16}; in TEST_F() local
57 std::vector<int> output_shape = {1, 2, 2, 16}; in TEST_F() local
74 std::vector<int> output_shape = {1, 2, 2, 36}; in TEST_F() local
101 std::vector<int> output_shape = {1, 1, 1, 4}; in TEST_F() local
114 std::vector<int> output_shape = {1, 1, 1, 12}; in TEST_F() local
129 std::vector<int> output_shape = {1, 2, 2, 12}; in TEST_F() local
145 std::vector<int> output_shape = {1, 2, 2, 54}; in TEST_F() local
178 std::vector<int> output_shape = {1, input_shape[1] * block_size, input_shape[2] * block_size, in TEST_F() local
196 std::vector<int> output_shape = {1, input_shape[1] * block_size, input_shape[2] * block_size, in TEST_F() local
212 std::vector<int> output_shape = {1, input_shape[1] * block_size, input_shape[2] * block_size, in TEST_F() local
[all …]
|
| D | softmax_tests.cc | 35 std::vector<int> output_shape = input_shape; variable
49 std::vector<int> output_shape = input_shape; variable
63 std::vector<int> output_shape = input_shape; variable
77 std::vector<int> output_shape = input_shape; variable
92 std::vector<int> output_shape = input_shape; variable
107 std::vector<int> output_shape = input_shape; variable
|
| D | one_hot_tests.cc | 39 std::vector<int> output_shape = input_shape; in TEST_F() local
59 std::vector<int> output_shape = input_shape; in TEST_F() local
80 std::vector<int> output_shape = input_shape; in TEST_F() local
103 std::vector<int> output_shape = input_shape; in TEST_F() local
126 std::vector<int> output_shape = input_shape; in TEST_F() local
147 std::vector<int> output_shape = input_shape; in TEST_F() local
169 std::vector<int> output_shape = input_shape; in TEST_F() local
190 std::vector<int> output_shape = input_shape; in TEST_F() local
213 std::vector<int> output_shape = input_shape; in TEST_F() local
236 std::vector<int> output_shape = input_shape; in TEST_F() local
[all …]
|
| D | argminmax_tests.cc | 44 std::vector<int> output_shape = {2, 2, 2, 2}; in TEST_F() local
59 std::vector<int> output_shape = {2, 2, 2, 2}; in TEST_F() local
74 std::vector<int> output_shape = {2, 2, 2, 3}; in TEST_F() local
90 std::vector<int> output_shape = {2, 2, 2, 3}; in TEST_F() local
107 std::vector<int> output_shape = {2, 3, 1, 3}; in TEST_F() local
124 std::vector<int> output_shape = {2, 2, 2, 5}; in TEST_F() local
142 std::vector<int> output_shape = {2, 2, 2, 5}; in TEST_F() local
160 std::vector<int> output_shape = {2, 2, 3, 2}; in TEST_F() local
177 std::vector<int> output_shape = {2, 2, 3, 2}; in TEST_F() local
194 std::vector<int> output_shape = {1, 14}; in TEST_F() local
[all …]
|
| D | arithmetic_tests.cc | 61 std::vector<int> output_shape = {1, 2, 2, 3}; in TEST_F() local
76 std::vector<int> output_shape = {1, 2, 2, 3}; in TEST_F() local
90 std::vector<int> output_shape = {1, 2, 2, 3}; in TEST_F() local
105 std::vector<int> output_shape = {1, 2, 2, 3}; in TEST_F() local
119 std::vector<int> output_shape = {2, 2, 2, 3}; in TEST_F() local
133 std::vector<int> output_shape = {1, 1, 3, 4}; in TEST_F() local
148 std::vector<int> output_shape = {1, 1, 3, 4}; in TEST_F() local
163 std::vector<int> output_shape = {1, 2, 2, 3}; in TEST_F() local
|
| D | reduce_tests.cc | 43 std::vector<int> output_shape = {1, 3}; in TEST_F() local
58 std::vector<int> output_shape = {1, 3}; in TEST_F() local
73 std::vector<int> output_shape = {1, 3, 1, 1}; in TEST_F() local
88 std::vector<int> output_shape = {1, 3, 1, 1}; in TEST_F() local
103 std::vector<int> output_shape = {1, 3, 2, 1}; in TEST_F() local
|
| D | activation_tests.cc | 37 std::vector<int> output_shape = input_shape; in TEST_F() local
48 std::vector<int> output_shape = input_shape; in TEST_F() local
59 std::vector<int> output_shape = input_shape; in TEST_F() local
77 std::vector<int> output_shape = input_shape; in TEST_F() local
|
| D | gather_tests.cc | 34 std::vector<int> output_shape = {5}; in TEST_F() local
53 std::vector<int> output_shape = {1}; in TEST_F() local
72 std::vector<int> output_shape = {1}; in TEST_F() local
91 std::vector<int> output_shape = {1, 2, 4, 4}; in TEST_F() local
122 std::vector<int> output_shape = {1, 2, 4, 4}; in TEST_F() local
153 std::vector<int> output_shape = {1, 5, 2, 4}; in TEST_F() local
176 std::vector<int> output_shape = {1, 5, 2, 4}; in TEST_F() local
199 std::vector<int> output_shape = {1, 5, 4, 2}; in TEST_F() local
222 std::vector<int> output_shape = {1, 5, 4, 2}; in TEST_F() local
|
| D | transpose_tests.cc | 40 std::vector<int> output_shape; in TEST_F() local
58 std::vector<int> output_shape; in TEST_F() local
76 std::vector<int> output_shape; in TEST_F() local
94 std::vector<int> output_shape; in TEST_F() local
112 std::vector<int> output_shape; in TEST_F() local
|
| D | concat_tests.cc | 35 std::vector<int> output_shape = {2, 1, 1, 8}; in TEST_F() local
50 std::vector<int> output_shape = {2}; in TEST_F() local
65 std::vector<int> output_shape = {2, 4, 2, 8}; in TEST_F() local
97 std::vector<int> output_shape = {2, 18, 2, 8}; in TEST_F() local
192 std::vector<int> output_shape = {1, 1, 48}; in TEST_F() local
219 std::vector<int> output_shape = {2, 2, 2, 17}; in TEST_F() local
251 std::vector<int> output_shape = {1, 21}; in TEST_F() local
270 std::vector<int> output_shape = {1, 1, 1, 30}; in TEST_F() local
294 std::vector<int> output_shape = {1, 1, 1, 40}; in TEST_F() local
322 std::vector<int> output_shape = {1, 1, 1, 51}; in TEST_F() local
|
| D | resize_tests.cc | 45 std::vector<int> output_shape = {1, oh, ow, 1}; in TEST_F() local
61 std::vector<int> output_shape = {1, oh, ow, 1}; in TEST_F() local
77 std::vector<int> output_shape = {1, oh, ow, 1}; in TEST_F() local
93 std::vector<int> output_shape = {2, oh, ow, 1}; in TEST_F() local
110 std::vector<int> output_shape = {2, oh, ow, 1}; in TEST_F() local
126 std::vector<int> output_shape = {2, oh, ow, 1}; in TEST_F() local
|
| D | sparse_to_dense_tests.cc | 36 std::vector<int> output_shape = {6, 1, 10}; in TEST_F() local
59 std::vector<int> output_shape = {6, 10}; in TEST_F() local
82 std::vector<int> output_shape = {6, 10}; in TEST_F() local
105 std::vector<int> output_shape = {10}; in TEST_F() local
127 std::vector<int> output_shape = {10}; in TEST_F() local
149 std::vector<int> output_shape = {10}; in TEST_F() local
171 std::vector<int> output_shape = {10}; in TEST_F() local
|
| D | matmul_tests.cc | 39 std::vector<int> output_shape = {1, co}; variable
59 std::vector<int> output_shape = {a, b, m, co}; variable
79 std::vector<int> output_shape = {a, m, co}; variable
99 std::vector<int> output_shape = {a, m, co}; in TEST_F() local
119 std::vector<int> output_shape = {a, m, co}; in TEST_F() local
|
| D | pooling_tests.cc | 48 std::vector<int> output_shape = {1, 1, 1, 4}; in TEST_F() local
59 std::vector<int> output_shape = {1, 1, 1, 4}; in TEST_F() local
70 std::vector<int> output_shape = {2, 1, 1, 4}; in TEST_F() local
82 std::vector<int> output_shape = {2, 1, 1, 4}; in TEST_F() local
|
| D | batch_to_space_nd_tests.cc | 27 std::vector<int> *output_shape) { in CreateParameter()
42 std::vector<int> output_shape; in TEST_F() local
86 std::vector<int> output_shape; in TEST_F() local
|
| D | power_tests.cc | 41 std::vector<int> output_shape = {1, 2, 8}; in TEST_F() local
59 std::vector<int> output_shape = {2, 8}; in TEST_F() local
79 std::vector<int> output_shape = {2, 7}; in TEST_F() local
97 std::vector<int> output_shape = {1, 2, 8}; in TEST_F() local
113 std::vector<int> output_shape = {2, 8}; in TEST_F() local
|
| D | stack_tests.cc | 37 std::vector<int> output_shape = {10, 2}; in TEST_F() local
54 std::vector<int> output_shape = {3, 2, 4}; in TEST_F() local
73 std::vector<int> output_shape = {3, 2, 4, 5}; in TEST_F() local
92 std::vector<int> output_shape = {3, 4, 2, 5}; in TEST_F() local
111 std::vector<int> output_shape = {1, 96, 2}; in TEST_F() local
130 std::vector<int> output_shape = {3, 4, 6, 2}; in TEST_F() local
|
| D | fullconnection_tests.cc | 26 … std::vector<int> *bias_shape, std::vector<int> *output_shape, int ndim, int ci, int co, in CreateParameter()
64 std::vector<int> input_shape, weight_shape, bias_shape, output_shape; variable
86 std::vector<int> input_shape, weight_shape, bias_shape, output_shape; variable
109 std::vector<int> input_shape, weight_shape, bias_shape, output_shape; variable
132 std::vector<int> input_shape, weight_shape, bias_shape, output_shape; variable
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | resize_nearest_neighbor_fp32_tests.cc | 50 …estNeighborFp32::Prepare(const std::vector<int> &input_shape, const std::vector<int> &output_shape, in Prepare()
78 std::vector<int> output_shape = {1, 1, 1, 1}; in TEST_F() local
95 std::vector<int> output_shape = {1, 1, 1, 1}; in TEST_F() local
112 std::vector<int> output_shape = {1, 1, 2, 1}; in TEST_F() local
129 std::vector<int> output_shape = {1, 2, 1, 1}; in TEST_F() local
146 std::vector<int> output_shape = {1, 2, 2, 1}; in TEST_F() local
163 std::vector<int> output_shape = {1, 1, 4, 1}; in TEST_F() local
180 std::vector<int> output_shape = {1, 4, 1, 1}; in TEST_F() local
197 std::vector<int> output_shape = {1, 2, 4, 1}; in TEST_F() local
214 std::vector<int> output_shape = {1, 4, 2, 1}; in TEST_F() local
[all …]
|
| D | resize_bilinear_fp32_tests.cc | 53 …izeBilinearFp32::Prepare(const std::vector<int> &input_shape, const std::vector<int> &output_shape, in Prepare()
83 std::vector<int> output_shape = {1, 1, 1, 1}; in TEST_F() local
100 std::vector<int> output_shape = {1, 1, 1, 1}; in TEST_F() local
117 std::vector<int> output_shape = {1, 1, 2, 1}; in TEST_F() local
134 std::vector<int> output_shape = {1, 2, 1, 1}; in TEST_F() local
151 std::vector<int> output_shape = {1, 2, 2, 1}; in TEST_F() local
168 std::vector<int> output_shape = {1, 1, 4, 1}; in TEST_F() local
185 std::vector<int> output_shape = {1, 4, 1, 1}; in TEST_F() local
202 std::vector<int> output_shape = {1, 2, 4, 1}; in TEST_F() local
219 std::vector<int> output_shape = {1, 4, 2, 1}; in TEST_F() local
[all …]
|
| D | pad_fp32_test.cc | 60 void TestPadFp32::Prepare(const std::vector<int> &input_shape, const std::vector<int> &output_shape… in Prepare()
99 std::vector<int> output_shape{1, 12, 12, 3}; in TEST_F() local
141 std::vector<int> output_shape{1, 10, 10, 3}; in TEST_F() local
176 std::vector<int> output_shape{1, 10, 10, 3}; in TEST_F() local
212 std::vector<int> output_shape{1, 10, 10, 3}; in TEST_F() local
248 std::vector<int> output_shape{10, 10, 3}; in TEST_F() local
|
| D | l2norm_fp32_test.cc | 49 void TestL2NormFp32::Init(const std::vector<int> &input_shape, const std::vector<int> &output_shape… in Init()
84 std::vector<int> output_shape = {1, 3, 2, 3}; in TEST_F() local
105 std::vector<int> output_shape = {1, 3, 2, 3}; in TEST_F() local
126 std::vector<int> output_shape = {1, 3, 2, 3}; in TEST_F() local
147 std::vector<int> output_shape = {1, 3, 2, 3}; in TEST_F() local
|
| D | one_hot_fp32_test.cc | 58 … const int axis, const std::vector<int> &output_shape, float *output_data, in Prepare()
88 std::vector<int> output_shape{3, 3, 4}; in TEST_F() local
107 std::vector<int> output_shape{3, 4, 3}; in TEST_F() local
126 std::vector<int> output_shape{4, 3, 3}; in TEST_F() local
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/ |
| D | reduce_int8_tests.cc | 101 std::vector<int> output_shape = {2, 4, 4, 1}; in TEST_F() local
126 std::vector<int> output_shape = {1}; in TEST_F() local
149 std::vector<int> output_shape = {2, 4, 4, 1}; in TEST_F() local
173 std::vector<int> output_shape = {1}; in TEST_F() local
195 std::vector<int> output_shape = {2, 4, 4, 1}; in TEST_F() local
218 std::vector<int> output_shape = {1}; in TEST_F() local
240 std::vector<int> output_shape = {2, 4, 4, 1}; in TEST_F() local
263 std::vector<int> output_shape = {1}; in TEST_F() local
286 std::vector<int> output_shape = {2, 4, 4, 1}; in TEST_F() local
307 std::vector<int> output_shape = {1, 2}; in TEST_F() local
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | dynamic_range_impl.cu | 24 int64_t *output_shape, DynamicRangeErrorCode *error_code, in ValidateInputAndInferShape()
65 …e(const T *range_start, const T *range_end, const T *range_delta, T *output, int64_t *output_shape, in Range()
78 int64_t *output_shape, DynamicRangeErrorCode *error_code, in CudaValidateInputAndInferShape()
85 …e(const T *range_start, const T *range_end, const T *range_delta, T *output, int64_t *output_shape, in CalRange()
|