| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | crop_fp32_test.cc | 32 int in_shape[4] = {2, 2, 2, 2}; in TEST_F() local
54 int in_shape[4] = {2, 2, 2, 2}; in TEST_F() local
77 int in_shape[4] = {2, 2, 2, 2}; in TEST_F() local
97 int in_shape[4] = {2, 2, 2, 2}; in TEST_F() local
118 int in_shape[4] = {2, 2, 2, 2}; in TEST_F() local
139 int in_shape[4] = {2, 2, 2, 2}; in TEST_F() local
161 int in_shape[4] = {2, 2, 2, 2}; in TEST_F() local
180 int in_shape[4] = {1, 3, 3, 3}; in TEST_F() local
206 int in_shape[4] = {1, 4, 4, 4}; in TEST_F() local
231 int in_shape[4] = {1, 2, 5, 5}; in TEST_F() local
[all …]
|
| D | batch_to_space_fp32_test.cc | 35 int in_shape[4] = {4, 1, 1, 3}; in TEST_F() local
52 int in_shape[4] = {4, 1, 1, 3}; in TEST_F() local
72 int in_shape[4] = {4, 2, 2, 2}; in TEST_F() local
90 int in_shape[4] = {4, 2, 2, 2}; in TEST_F() local
114 int in_shape[4] = {8, 2, 2, 2}; in TEST_F() local
134 int in_shape[4] = {8, 2, 2, 2}; in TEST_F() local
160 int in_shape[4] = {12, 2, 2, 2}; in TEST_F() local
182 int in_shape[4] = {12, 2, 2, 2}; in TEST_F() local
|
| D | reduce_fp32_tests.cc | 70 void TestReduceFp32::Prepare(const std::vector<int> &in_shape, const std::vector<int> &out_shape, f… in Prepare()
110 std::vector<int> in_shape{2, 4, 4, 3}; in TEST_F() local
137 std::vector<int> in_shape{2, 4, 4, 3}; in TEST_F() local
165 std::vector<int> in_shape{2, 4, 4, 3}; in TEST_F() local
192 std::vector<int> in_shape{2, 4, 4, 3}; in TEST_F() local
218 std::vector<int> in_shape{2, 4, 4, 3}; in TEST_F() local
247 std::vector<int> in_shape{2, 4, 4, 3}; in TEST_F() local
275 std::vector<int> in_shape{2, 4, 4, 3}; in TEST_F() local
301 std::vector<int> in_shape{2, 4, 4, 3}; in TEST_F() local
327 std::vector<int> in_shape{2, 4, 4, 3}; in TEST_F() local
[all …]
|
| D | strided_slice_fp32_tests.cc | 32 void InitStridedSliceParam(StridedSliceParameter *param, const std::vector<int> &in_shape, in InitStridedSliceParam()
48 std::vector<int> in_shape{1, 2, 4}; in TEST_F() local
97 std::vector<int> in_shape{1, 3, 3}; in TEST_F() local
122 std::vector<int> in_shape{1, 2, 4}; in TEST_F() local
177 std::vector<int> in_shape{5, 5}; in TEST_F() local
231 std::vector<int> in_shape{5, 5, 5}; in TEST_F() local
292 std::vector<int> in_shape{5, 5, 5}; in TEST_F() local
353 std::vector<int> in_shape{2, 3}; in TEST_F() local
406 std::vector<int> in_shape{5, 5, 5}; in TEST_F() local
468 std::vector<int> in_shape{2, 1, 10, 7, 7}; in TEST_F() local
|
| D | space_to_batch_fp32_tests.cc | 36 std::vector<int> in_shape = {1, 4, 4, 1}; in TEST_F() local
57 std::vector<int> in_shape = {1, 4, 4, 1}; in TEST_F() local
78 std::vector<int> in_shape = {1, 4, 4, 1}; in TEST_F() local
|
| D | depth_to_space_fp32_test.cc | 34 int in_shape[4] = {1, 2, 2, 4}; in TEST_F() local
63 int in_shape[4] = {1, 1, 1, 8}; in TEST_F() local
|
| D | space_to_depth_fp32_tests.cc | 39 int in_shape[4] = {1, 4, 4, 1}; in TEST_F() local
53 std::vector<int> in_shape = {1, 4, 4, 1}; in TEST_F() local
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/ |
| D | resize_nearest_neighbor_int8_tests.cc | 50 void TestResizeNearestNeighborInt8::Prepare(const std::vector<int> &in_shape, const std::vector<int… in Prepare()
91 std::vector<int> in_shape = {1, 2, 2, 1}; in TEST_F() local
111 std::vector<int> in_shape = {2, 2, 2, 5}; in TEST_F() local
137 std::vector<int> in_shape = {2, 2, 2, 5}; in TEST_F() local
164 std::vector<int> in_shape = {2, 2, 2, 5}; in TEST_F() local
190 std::vector<int> in_shape = {2, 2, 2, 5}; in TEST_F() local
|
| D | resize_bilinear_int8_tests.cc | 55 void TestResizeBilinearInt8::Prepare(const std::vector<int> &in_shape, const std::vector<int> &out_… in Prepare()
91 std::vector<int> in_shape = {1, 2, 2, 1}; in TEST_F() local
107 std::vector<int> in_shape = {2, 2, 2, 5}; in TEST_F() local
132 std::vector<int> in_shape = {2, 2, 2, 5}; in TEST_F() local
158 std::vector<int> in_shape = {2, 2, 2, 5}; in TEST_F() local
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/base/ |
| D | softmax_base_coder.cc | 23 std::vector<int> in_shape = input_tensor_->shape(); in Init() local
38 std::vector<int> in_shape = input_tensor_->shape(); in ReSize() local
|
| D | strided_slice_base_coder.cc | 45 std::vector<int> in_shape = input_tensor_->shape(); in MatchFastPattern() local
64 std::vector<int> in_shape = input_tensor_->shape(); in InitFastRunParam() local
100 std::vector<int> in_shape = input_tensor_->shape(); in DoFastCode() local
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/ |
| D | reduce_int8_coder.cc | 42 std::vector<int> in_shape = input_tensor_->shape(); in CalculateQuantArgs() local
60 std::vector<int> in_shape = input_tensors_.at(kInputIndex)->shape(); in CalculateQuantArgs() local
139 std::vector<int> in_shape = input_tensor_->shape(); in Prepare() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | argminmax_int8.cc | 75 auto in_shape = in_tensors_.at(0)->shape(); in ReSize() local
103 auto in_shape = input->shape(); in Run() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | strided_slice.cc | 47 auto in_shape = in_tensors_.front()->shape(); in InitFastRunParam() local
95 auto in_shape = in_tensors_.front()->shape(); in MatchFastPattern() local
114 auto in_shape = in_tensors_.front()->shape(); in FastRunImpl() local
|
| D | stack_base.cc | 36 static int GetCopyNum(const std::vector<int> &in_shape, int axis, int n_dim) { in GetCopyNum()
50 static inline int GetOuterSize(const std::vector<int> &in_shape, int axis) { in GetOuterSize()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | strided_slice_infer.c | 109 int HandleAxesInputExist(const TensorC *const *inputs, int *ndim, int *in_shape, int *begins, int *… in HandleAxesInputExist()
212 …wAxisMask(StridedSliceTransferBuffer *transfer_buffer, StridedSliceParameter *param, int *in_shape, in ApplyNewAxisMask()
248 int ApplyEndMask(StridedSliceTransferBuffer *transfer_buffer, const int *in_shape, size_t in_shape_… in ApplyEndMask()
260 int ApplyEllipsisMask(StridedSliceTransferBuffer *transfer_buffer, const int *in_shape, size_t in_s… in ApplyEllipsisMask()
274 int TransIndexToPositive(StridedSliceTransferBuffer *transfer_buffer, const int *in_shape, size_t i… in TransIndexToPositive()
308 const int *in_shape, size_t in_shape_size) { in TransferBuffer2Param()
361 int in_shape[MAX_SHAPE_SIZE] = {0}; in StridedSliceInferShape() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/ |
| D | cholesky_solve_gpu_kernel.h | 118 auto in_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0); in Init() local
139 bool InitNoSpltDim(const std::vector<size_t> &in_shape) { in InitNoSpltDim()
177 bool InitSpltDim(const std::vector<size_t> &in_shape) { in InitSpltDim()
|
| D | cholesky_trsm_solve_gpu_kernel.h | 69 auto in_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0); in Init() local
195 bool InitDim0(const CNodePtr &kernel_node, const std::vector<size_t> &in_shape) { in InitDim0()
231 void InitDimOthers(const CNodePtr &kernel_node, const std::vector<size_t> &in_shape) { in InitDimOthers()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/ |
| D | argminmax.cc | 67 …cl_int4 in_shape{static_cast<int>(im_in_.N), static_cast<int>(im_in_.H), static_cast<int>(im_in_.W… in SetConstArgs() local
107 std::vector<size_t> in_shape = {im_in_.N, im_in_.H, im_in_.W, im_in_.C}; in SetGlobalLocalPre() local
|
| D | resize.cc | 38 auto in_shape = in_tensors_[0]->shape(); in CheckSpecs() local
94 auto in_shape = in_tensors_[0]->shape(); in SetConstArgs() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | im2col_gpu_kernel.h | 90 auto in_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0); in Init() local
206 void SetPad(const std::vector<size_t> &in_shape, const CNodePtr &kernel_node) { in SetPad()
241 void Set4DDesc(const std::vector<size_t> &in_shape, const std::vector<int> &filter_shape, in Set4DDesc()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | arg_min_max_fp16.c | 105 …gMinMaxDim0Fp16(const float16_t *input, void *output, float16_t *output_value, const int *in_shape, in ArgMinMaxDim0Fp16()
131 …gMinMaxDim1Fp16(const float16_t *input, void *output, float16_t *output_value, const int *in_shape, in ArgMinMaxDim1Fp16()
162 …axDim2Fp16(const float16_t *input, float16_t *output, float16_t *output_value, const int *in_shape, in ArgMinMaxDim2Fp16()
197 …axDim3Fp16(const float16_t *input, float16_t *output, float16_t *output_value, const int *in_shape, in ArgMinMaxDim3Fp16()
235 …d ArgMinMaxFp16(const float16_t *input, void *output, float16_t *output_value, const int *in_shape, in ArgMinMaxFp16()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | arg_min_max_int8.c | 70 void Int8ArgMinMaxQuant(const int8_t *input, int8_t *output, const int *in_shape, const ArgMinMaxPa… in Int8ArgMinMaxQuant()
92 void Int8ArgMinMaxDim0(const int8_t *input, int8_t *output, const int *in_shape, ArgMinMaxParameter… in Int8ArgMinMaxDim0()
118 void Int8ArgMinMaxDim1(const int8_t *input, int8_t *output, const int *in_shape, ArgMinMaxParameter… in Int8ArgMinMaxDim1()
149 void Int8ArgMinMaxDim2(const int8_t *input, int8_t *output, const int *in_shape, ArgMinMaxParameter… in Int8ArgMinMaxDim2()
184 void Int8ArgMinMaxDim3(const int8_t *input, int8_t *output, const int *in_shape, ArgMinMaxParameter… in Int8ArgMinMaxDim3()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | arg_min_max_fp32.c | 106 void ArgMinMaxDim0(const float *input, void *output, float *output_value, const int *in_shape, in ArgMinMaxDim0()
132 void ArgMinMaxDim1(const float *input, void *output, float *output_value, const int *in_shape, in ArgMinMaxDim1()
163 void ArgMinMaxDim2(const float *input, void *output, float *output_value, const int *in_shape, in ArgMinMaxDim2()
198 void ArgMinMaxDim3(const float *input, void *output, float *output_value, const int *in_shape, in ArgMinMaxDim3()
236 void ArgMinMaxFp32(const float *input, void *output, float *output_value, const int *in_shape, in ArgMinMaxFp32()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | rank_fp32.cc | 41 auto in_shape = in_tensors_.at(0)->shape(); in Run() local
|