| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/_op_impl/_custom_op/ |
| D | img2col_impl.py | 39 def shape56_0(tik_instance, input_x, res, input_shape, shape_info): argument
66 def shape56_1(tik_instance, input_x, res, input_shape, shape_info): argument
94 def shape56_2(tik_instance, input_x, res, input_shape, shape_info): argument
118 def shape56_3(tik_instance, input_x, res, input_shape, shape_info): argument
146 def shape56_4(tik_instance, input_x, res, input_shape, shape_info): argument
175 def shape28_0(tik_instance, input_x, res, input_shape, shape_info): argument
203 def shape28_1(tik_instance, input_x, res, input_shape, shape_info): argument
242 def shape28_2(tik_instance, input_x, res, input_shape, shape_info): argument
290 def shape28_3(tik_instance, input_x, res, input_shape, shape_info): argument
318 def shape28_4(tik_instance, input_x, res, input_shape, shape_info): argument
[all …]
|
| /third_party/mindspore/mindspore-src/source/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 | 41 std::vector<int> input_shape = {3, 2, 2, 2}; in TEST_F() local
56 std::vector<int> input_shape = {3, 2, 2, 2}; in TEST_F() local
71 std::vector<int> input_shape = {2, 3, 2, 3}; in TEST_F() local
87 std::vector<int> input_shape = {2, 3, 2, 3}; in TEST_F() local
104 std::vector<int> input_shape = {2, 3, 3, 3}; in TEST_F() local
121 std::vector<int> input_shape = {2, 2, 3, 5}; in TEST_F() local
139 std::vector<int> input_shape = {2, 2, 3, 5}; in TEST_F() local
157 std::vector<int> input_shape = {2, 2, 3, 5}; in TEST_F() local
174 std::vector<int> input_shape = {2, 2, 3, 5}; in TEST_F() local
191 std::vector<int> input_shape = {1, 2, 14}; in TEST_F() local
[all …]
|
| D | reduce_tests.cc | 38 std::vector<int> input_shape = {1, 2, 2, 3}; in TEST_F() local
53 std::vector<int> input_shape = {1, 2, 2, 3}; in TEST_F() local
68 std::vector<int> input_shape = {1, 3, 2, 2}; in TEST_F() local
83 std::vector<int> input_shape = {1, 3, 2, 2}; in TEST_F() local
98 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 = {5, 4, 4}; in TEST_F() local
151 std::vector<int> input_shape = {1, 5, 4, 4}; in TEST_F() local
182 std::vector<int> input_shape = {1, 5, 4, 4}; in TEST_F() local
205 std::vector<int> input_shape = {1, 5, 4, 4}; in TEST_F() local
228 std::vector<int> input_shape = {1, 5, 4, 4}; in TEST_F() local
251 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
|
| /third_party/mindspore/mindspore-src/source/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 | 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-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/ |
| D | reduce_int8_tests.cc | 107 std::vector<int> input_shape = {2, 4, 4, 3}; in TEST_F() local
132 std::vector<int> input_shape = {2, 4, 4, 3}; in TEST_F() local
155 std::vector<int> input_shape = {2, 4, 4, 3}; in TEST_F() local
179 std::vector<int> input_shape = {2, 4, 4, 3}; in TEST_F() local
201 std::vector<int> input_shape = {2, 4, 4, 3}; in TEST_F() local
224 std::vector<int> input_shape = {2, 4, 4, 3}; in TEST_F() local
246 std::vector<int> input_shape = {2, 4, 4, 3}; in TEST_F() local
269 std::vector<int> input_shape = {2, 4, 4, 3}; in TEST_F() local
292 std::vector<int> input_shape = {2, 4, 4, 3}; in TEST_F() local
313 std::vector<int> input_shape = {1, 2, 2, 3}; in TEST_F() local
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/checker/ |
| D | resize_checker.cc | 40 bool CheckInterpOp(const api::CNodePtr &op, const ShapeVector &input_shape, const ShapeVector &outp… in CheckInterpOp()
77 bool IsDoubleResize(const api::CNodePtr &, const ShapeVector &input_shape, const ShapeVector &outpu… in IsDoubleResize()
83 bool CheckResizeOp(const api::CNodePtr &op, const ShapeVector &input_shape, const ShapeVector &outp… in CheckResizeOp()
123 ShapeVector input_shape; in Check() local
|
| D | reduce_checker.cc | 30 STATUS GetAxesSet(const api::CNodePtr &op, const ShapeVector &input_shape, const api::PrimitivePtr … in GetAxesSet()
65 const ShapeVector &input_shape) { in CheckAttr()
107 std::vector<int64_t> input_shape; in Check() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | unstack_ext.cc | 33 … CheckShapeAndGetNormalizedAxisValue(const PrimitivePtr &primitive, const ShapeVector &input_shape, in CheckShapeAndGetNormalizedAxisValue()
58 auto input_shape = input_args[kInputIndex0]->GetShape()->GetShapeVector(); in InferShape() local
83 auto input_shape = input_args[kInputIndex0]->GetShape()->GetShapeVector(); in InferType() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| 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-src/source/mindspore/lite/examples/cloud_infer/runtime_cpp/ |
| D | config_file.ini | 2 input_shape=input_1:[-1,3,224,224] key
6 input_shape=input_1:[-1,3,224,224] key
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/ |
| D | ops_utils.cc | 41 std::vector<int64_t> *stride, ShapeVector *input_shape, bool is_gpu_strided) { in FillEmptyDims()
86 …sk(std::vector<int64_t> *begin, const std::vector<int64_t> &stride, const ShapeVector &input_shape, in ComputeBeginMask()
98 …Mask(std::vector<int64_t> *end, const std::vector<int64_t> &stride, const ShapeVector &input_shape, in ComputeEndMask()
111 const ShapeVector &input_shape, KernelTensor *kernel_tensor) { in ComputeEllipsisMask()
127 const ShapeVector &input_shape, KernelTensor *kernel_tensor) { in ComputNewAxisMask()
157 … std::vector<int64_t> *end, std::vector<int64_t> *stride, const ShapeVector &input_shape) { in ParseStrideSliceMasks()
167 std::vector<int64_t> *stride, ShapeVector *input_shape, bool is_gpu_strided) { in FillEmptyDims()
239 const std::vector<int64_t> &step, const std::vector<int64_t> &input_shape) { in CheckSliceValid()
282 std::vector<int64_t> CalDimOffset(const std::vector<int64_t> &input_shape) { in CalDimOffset()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/parallel/ |
| D | multi_conv_info.h | 76 …d AdJustConvPrim(const api::SharedPtr<ops::Conv2DFusion> &ori_attr, const ShapeVector &input_shape, in AdJustConvPrim()
87 …d AdJustConvPrim(const api::SharedPtr<ops::Conv2DFusion> &ori_attr, const ShapeVector &input_shape, in AdJustConvPrim()
99 …d AdJustConvPrim(const api::SharedPtr<ops::Conv2DFusion> &ori_attr, const ShapeVector &input_shape, in AdJustConvPrim()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/text/kernels/ |
| D | sliding_window_op.cc | 29 std::vector<TensorShape> input_shape = {input->shape()}; in Compute() local
40 TensorShape input_shape = inputs[0]; in OutputShape() local
|