| /third_party/mindspore/mindspore/ops/_op_impl/_custom_op/ |
| D | img2col_impl.py | 37 def shape56_0(tik_instance, input_x, res, input_shape, shape_info): argument
64 def shape56_1(tik_instance, input_x, res, input_shape, shape_info): argument
92 def shape56_2(tik_instance, input_x, res, input_shape, shape_info): argument
116 def shape56_3(tik_instance, input_x, res, input_shape, shape_info): argument
144 def shape56_4(tik_instance, input_x, res, input_shape, shape_info): argument
173 def shape28_0(tik_instance, input_x, res, input_shape, shape_info): argument
201 def shape28_1(tik_instance, input_x, res, input_shape, shape_info): argument
240 def shape28_2(tik_instance, input_x, res, input_shape, shape_info): argument
288 def shape28_3(tik_instance, input_x, res, input_shape, shape_info): argument
316 def shape28_4(tik_instance, input_x, res, input_shape, shape_info): argument
[all …]
|
| D | batch_matmul_impl.py | 46 def _error_feedback(input_shape): argument
118 def process_input_shape_640(input_shape, tik_instance, dtype, total_input, res): argument
169 def process_input_shape_1152(input_shape, tik_instance, dtype, total_input, res): argument
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | space_to_depth_tests.cc | 43 std::vector<int> input_shape = {1, 2, 2, 4}; in TEST_F() local
56 std::vector<int> input_shape = {1, 4, 4, 4}; in TEST_F() local
73 std::vector<int> input_shape = {1, 6, 6, 4}; in TEST_F() local
100 std::vector<int> input_shape = {1, 2, 2, 1}; in TEST_F() local
113 std::vector<int> input_shape = {1, 2, 2, 3}; in TEST_F() local
128 std::vector<int> input_shape = {1, 4, 4, 3}; in TEST_F() local
144 std::vector<int> input_shape = {1, 6, 6, 6}; in TEST_F() local
177 std::vector<int> input_shape = {1, 2, 2, 16}; in TEST_F() local
195 std::vector<int> input_shape = {1, 1, 2, 16}; in TEST_F() local
211 std::vector<int> input_shape = {1, 1, 2, 36}; in TEST_F() local
[all …]
|
| D | softmax_tests.cc | 34 std::vector<int> input_shape = {1, 10}; variable
48 std::vector<int> input_shape = {1, 2, 1, 5}; variable
62 std::vector<int> input_shape = {1, 2, 1, 1}; variable
76 std::vector<int> input_shape = {2, 10}; variable
91 std::vector<int> input_shape = {2, 2, 1, 5}; variable
106 std::vector<int> input_shape = {2, 2, 1, 1}; variable
|
| D | one_hot_tests.cc | 38 std::vector<int> input_shape = {1, 2, 2}; in TEST_F() local
58 std::vector<int> input_shape = {1, 2, 2}; in TEST_F() local
79 std::vector<int> input_shape = {1, 2, 3}; in TEST_F() local
102 std::vector<int> input_shape = {1, 2, 5}; in TEST_F() local
125 std::vector<int> input_shape = {1, 2, 2}; in TEST_F() local
146 std::vector<int> input_shape = {1, 6, 2}; in TEST_F() local
168 std::vector<int> input_shape = {1, 2, 2}; in TEST_F() local
189 std::vector<int> input_shape = {1, 2, 5}; in TEST_F() local
212 std::vector<int> input_shape = {1, 6, 6}; in TEST_F() local
235 std::vector<int> input_shape = {1, 2, 2}; in TEST_F() local
[all …]
|
| D | argminmax_tests.cc | 43 std::vector<int> input_shape = {3, 2, 2, 2}; in TEST_F() local
58 std::vector<int> input_shape = {3, 2, 2, 2}; in TEST_F() local
73 std::vector<int> input_shape = {2, 3, 2, 3}; in TEST_F() local
89 std::vector<int> input_shape = {2, 3, 2, 3}; in TEST_F() local
106 std::vector<int> input_shape = {2, 3, 3, 3}; in TEST_F() local
123 std::vector<int> input_shape = {2, 2, 3, 5}; in TEST_F() local
141 std::vector<int> input_shape = {2, 2, 3, 5}; in TEST_F() local
159 std::vector<int> input_shape = {2, 2, 3, 5}; in TEST_F() local
176 std::vector<int> input_shape = {2, 2, 3, 5}; in TEST_F() local
193 std::vector<int> input_shape = {1, 2, 14}; in TEST_F() local
[all …]
|
| D | reduce_tests.cc | 42 std::vector<int> input_shape = {1, 2, 2, 3}; in TEST_F() local
57 std::vector<int> input_shape = {1, 2, 2, 3}; in TEST_F() local
72 std::vector<int> input_shape = {1, 3, 2, 2}; in TEST_F() local
87 std::vector<int> input_shape = {1, 3, 2, 2}; in TEST_F() local
102 std::vector<int> input_shape = {1, 3, 2, 2}; in TEST_F() local
|
| D | activation_tests.cc | 36 std::vector<int> input_shape = {1, 2, 2, 3}; in TEST_F() local
47 std::vector<int> input_shape = {1, 2, 2, 3}; in TEST_F() local
58 std::vector<int> input_shape = {2, 10, 1, 4}; in TEST_F() local
76 std::vector<int> input_shape = {2, 10, 1, 4}; in TEST_F() local
|
| D | gather_tests.cc | 32 std::vector<int> input_shape = {10}; in TEST_F() local
51 std::vector<int> input_shape = {10}; in TEST_F() local
70 std::vector<int> input_shape = {10}; in TEST_F() local
89 std::vector<int> input_shape = {1, 5, 4, 4}; in TEST_F() local
120 std::vector<int> input_shape = {1, 5, 4, 4}; in TEST_F() local
151 std::vector<int> input_shape = {1, 5, 4, 4}; in TEST_F() local
174 std::vector<int> input_shape = {1, 5, 4, 4}; in TEST_F() local
197 std::vector<int> input_shape = {1, 5, 4, 4}; in TEST_F() local
220 std::vector<int> input_shape = {1, 5, 4, 4}; in TEST_F() local
|
| D | transpose_tests.cc | 38 std::vector<int> input_shape = {1, 2, 2, 3}; in TEST_F() local
56 std::vector<int> input_shape = {1, 2, 2, 3}; in TEST_F() local
74 std::vector<int> input_shape = {1, 2, 3, 4}; in TEST_F() local
92 std::vector<int> input_shape = {1, 2, 3}; in TEST_F() local
110 std::vector<int> input_shape = {2, 3, 5}; in TEST_F() local
|
| D | resize_tests.cc | 44 std::vector<int> input_shape = {1, 2, 2, 1}; in TEST_F() local
60 std::vector<int> input_shape = {1, 2, 2, 1}; in TEST_F() local
76 std::vector<int> input_shape = {1, 2, 2, 1}; in TEST_F() local
92 std::vector<int> input_shape = {2, 2, 2, 1}; in TEST_F() local
109 std::vector<int> input_shape = {2, 2, 2, 1}; in TEST_F() local
125 std::vector<int> input_shape = {2, 2, 2, 1}; in TEST_F() local
|
| D | matmul_tests.cc | 38 std::vector<int> input_shape = {1, ci}; variable
58 std::vector<int> input_shape = {a, b, m, ci}; variable
78 std::vector<int> input_shape = {a, m, ci}; variable
98 std::vector<int> input_shape = {a, m, ci}; in TEST_F() local
118 std::vector<int> input_shape = {a, m, ci}; in TEST_F() local
|
| D | pooling_tests.cc | 47 std::vector<int> input_shape = {1, 2, 2, 4}; in TEST_F() local
58 std::vector<int> input_shape = {1, 2, 2, 4}; in TEST_F() local
69 std::vector<int> input_shape = {2, 2, 2, 4}; in TEST_F() local
81 std::vector<int> input_shape = {2, 2, 2, 4}; in TEST_F() local
|
| D | batch_to_space_nd_tests.cc | 26 OpParameter *CreateParameter(int block_shape[], int crops[], const std::vector<int> &input_shape, in CreateParameter()
39 std::vector<int> input_shape = {4, 5, 5, 4}; in TEST_F() local
83 std::vector<int> input_shape = {9, 3, 3, 4}; in TEST_F() local
|
| D | fullconnection_tests.cc | 25 OpParameter *CreateParameter(std::vector<int> *input_shape, std::vector<int> *weight_shape, 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 void TestResizeNearestNeighborFp32::Prepare(const std::vector<int> &input_shape, const std::vector<… in Prepare()
77 std::vector<int> input_shape = {1, 1, 1, 1}; in TEST_F() local
94 std::vector<int> input_shape = {1, 2, 2, 1}; in TEST_F() local
111 std::vector<int> input_shape = {1, 2, 2, 1}; in TEST_F() local
128 std::vector<int> input_shape = {1, 2, 2, 1}; in TEST_F() local
145 std::vector<int> input_shape = {1, 2, 2, 1}; in TEST_F() local
162 std::vector<int> input_shape = {1, 2, 2, 1}; in TEST_F() local
179 std::vector<int> input_shape = {1, 2, 2, 1}; in TEST_F() local
196 std::vector<int> input_shape = {1, 2, 2, 1}; in TEST_F() local
213 std::vector<int> input_shape = {1, 2, 2, 1}; in TEST_F() local
[all …]
|
| D | resize_bilinear_fp32_tests.cc | 53 void TestResizeBilinearFp32::Prepare(const std::vector<int> &input_shape, const std::vector<int> &o… in Prepare()
82 std::vector<int> input_shape = {1, 1, 1, 1}; in TEST_F() local
99 std::vector<int> input_shape = {1, 2, 2, 1}; in TEST_F() local
116 std::vector<int> input_shape = {1, 2, 2, 1}; in TEST_F() local
133 std::vector<int> input_shape = {1, 2, 2, 1}; in TEST_F() local
150 std::vector<int> input_shape = {1, 2, 2, 1}; in TEST_F() local
167 std::vector<int> input_shape = {1, 2, 2, 1}; in TEST_F() local
184 std::vector<int> input_shape = {1, 2, 2, 1}; in TEST_F() local
201 std::vector<int> input_shape = {1, 2, 2, 1}; in TEST_F() local
218 std::vector<int> input_shape = {1, 2, 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()
98 std::vector<int> input_shape{1, 4, 4, 3}; in TEST_F() local
140 std::vector<int> input_shape{1, 4, 4, 3}; in TEST_F() local
175 std::vector<int> input_shape{1, 4, 4, 3}; in TEST_F() local
211 std::vector<int> input_shape{1, 4, 4, 3}; in TEST_F() local
247 std::vector<int> input_shape{4, 4, 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()
83 std::vector<int> input_shape = {1, 3, 2, 3}; in TEST_F() local
104 std::vector<int> input_shape = {1, 3, 2, 3}; in TEST_F() local
125 std::vector<int> input_shape = {1, 3, 2, 3}; in TEST_F() local
146 std::vector<int> input_shape = {1, 3, 2, 3}; in TEST_F() local
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/ |
| D | reduce_int8_tests.cc | 100 std::vector<int> input_shape = {2, 4, 4, 3}; in TEST_F() local
125 std::vector<int> input_shape = {2, 4, 4, 3}; in TEST_F() local
148 std::vector<int> input_shape = {2, 4, 4, 3}; in TEST_F() local
172 std::vector<int> input_shape = {2, 4, 4, 3}; in TEST_F() local
194 std::vector<int> input_shape = {2, 4, 4, 3}; in TEST_F() local
217 std::vector<int> input_shape = {2, 4, 4, 3}; in TEST_F() local
239 std::vector<int> input_shape = {2, 4, 4, 3}; in TEST_F() local
262 std::vector<int> input_shape = {2, 4, 4, 3}; in TEST_F() local
285 std::vector<int> input_shape = {2, 4, 4, 3}; in TEST_F() local
306 std::vector<int> input_shape = {1, 2, 2, 3}; in TEST_F() local
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | softmax_gpu_kernel.h | 73 size_t *input_shape = GetDeviceAddress<size_t>(workspace, 2); in Launch() local
116 auto input_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0); in Init() local
183 void InitSizeByAxis(const std::vector<size_t> &input_shape, const int &axis) { in InitSizeByAxis()
200 void InitSizeByAxis2D(const std::vector<size_t> &input_shape, const int &axis) { in InitSizeByAxis2D()
230 void InitSizeByAxisLastDim(const std::vector<size_t> &input_shape, const int &axis) { in InitSizeByAxisLastDim()
256 void InitSizeByAxisND(const std::vector<size_t> &input_shape, const int &axis) { in InitSizeByAxisND()
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_spacetobatch_op.py | 26 def __init__(self, nptype, block_size=2, input_shape=(1, 1, 4, 4)): argument
42 def SpaceToBatch(nptype, block_size=2, input_shape=(1, 1, 4, 4)): argument
61 def SpaceToBatch_pynative(nptype, block_size=2, input_shape=(1, 1, 4, 4)): argument
|
| D | test_batchtospace_op.py | 26 def __init__(self, nptype, block_size=2, input_shape=(4, 1, 2, 2)): argument
42 def BatchToSpace(nptype, block_size=2, input_shape=(4, 1, 2, 2)): argument
57 def BatchToSpace_pynative(nptype, block_size=2, input_shape=(4, 1, 2, 2)): argument
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/text/kernels/ |
| D | sliding_window_op.cc | 26 std::vector<TensorShape> input_shape = {input->shape()}; in Compute() local
37 TensorShape input_shape = inputs[0]; in OutputShape() local
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/parallel/ |
| D | multi_conv_info.h | 78 … AdJustConvPrim(const std::shared_ptr<ops::Conv2DFusion> &ori_attr, const ShapeVector &input_shape, in AdJustConvPrim()
89 … AdJustConvPrim(const std::shared_ptr<ops::Conv2DFusion> &ori_attr, const ShapeVector &input_shape, in AdJustConvPrim()
101 … AdJustConvPrim(const std::shared_ptr<ops::Conv2DFusion> &ori_attr, const ShapeVector &input_shape, in AdJustConvPrim()
|