| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | scatter_nd_functor_gpu_kernel.h | 52 const size_t indices_len = sizeof(S) * out_strides_.size(); in Launch()
56 … cudaMemcpyAsync(indices_stride, &out_strides_[0], indices_len, cudaMemcpyHostToDevice, in Launch()
131 out_strides_.push_back(out_stride); in Init()
134 out_strides_.push_back(out_stride); in Init()
136 reverse(out_strides_.begin(), out_strides_.end()); in Init()
159 workspace_size_list_.push_back(sizeof(S) * out_strides_.size()); in InitSizeLists()
171 std::vector<S> out_strides_; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | scatter_nd_cpu_kernel.cc | 34 std::vector<int> *out_strides = params->out_strides_; in Compute()
87 out_strides_.push_back(out_stride); in InitKernel()
90 out_strides_.push_back(out_stride); in InitKernel()
92 reverse(out_strides_.begin(), out_strides_.end()); in InitKernel()
112 params.out_strides_ = &out_strides_; in Launch()
|
| D | scatter_nd_update_cpu_kernel.cc | 35 std::vector<int> *out_strides = params->out_strides_; in Compute()
91 out_strides_.push_back(out_stride); in InitKernel()
94 out_strides_.push_back(out_stride); in InitKernel()
96 reverse(out_strides_.begin(), out_strides_.end()); in InitKernel()
126 params.out_strides_ = &out_strides_; in LaunchKernel()
|
| D | scatter_nd_update_cpu_kernel.h | 35 std::vector<int> *out_strides_{nullptr};
57 std::vector<int> out_strides_; variable
|
| D | transpose_cpu_kernel.cc | 54 transpose_param_.out_strides_[num_axes - 1] = 1; in InitKernel()
57 …transpose_param_.out_strides_[i] = SizeToInt(output_shape_[i + 1]) * transpose_param_.out_strides_… in InitKernel()
|
| D | scatter_nd_cpu_kernel.h | 34 std::vector<int> *out_strides_{nullptr};
55 std::vector<int> out_strides_; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | arg_min_max_int8.c | 111 int out_offset = j * param->out_strides_[0] + i; in Int8ArgMinMaxDim0()
127 int out_dim0_offset = i * param->out_strides_[0]; in Int8ArgMinMaxDim1()
141 int out_offset = out_dim0_offset + j + k * param->out_strides_[1]; in Int8ArgMinMaxDim1()
159 int out_dim0_offset = i * param->out_strides_[0]; in Int8ArgMinMaxDim2()
162 int out_dim1_offset = j * param->out_strides_[1] + out_dim0_offset; in Int8ArgMinMaxDim2()
175 int out_offset = out_dim1_offset + k + l * param->out_strides_[2]; in Int8ArgMinMaxDim2()
195 int out_dim0_offset = i * param->out_strides_[0]; in Int8ArgMinMaxDim3()
198 int out_dim1_offset = j * param->out_strides_[1] + out_dim0_offset; in Int8ArgMinMaxDim3()
201 int out_dim2_offset = k * param->out_strides_[2] + out_dim1_offset; in Int8ArgMinMaxDim3()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | arg_min_max_fp32.c | 118 int out_offset = j * param->out_strides_[0] + i; in ArgMinMaxDim0()
139 int out_dim0_offset = i * param->out_strides_[0]; in ArgMinMaxDim1()
148 int out_offset = out_dim0_offset + j + k * param->out_strides_[1]; in ArgMinMaxDim1()
171 int out_dim0_offset = i * param->out_strides_[0]; in ArgMinMaxDim2()
174 int out_dim1_offset = j * param->out_strides_[1] + out_dim0_offset; in ArgMinMaxDim2()
183 int out_offset = out_dim1_offset + k + l * param->out_strides_[2]; in ArgMinMaxDim2()
207 int out_dim0_offset = i * param->out_strides_[0]; in ArgMinMaxDim3()
210 int out_dim1_offset = j * param->out_strides_[1] + out_dim0_offset; in ArgMinMaxDim3()
213 int out_dim2_offset = k * param->out_strides_[2] + out_dim1_offset; in ArgMinMaxDim3()
|
| D | roi_pooling_fp32.c | 77 …int pooled_index = i * param->out_strides_[0] + ph * param->out_strides_[1] + pw * param->out_stri… in ROIPooling()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | tile_fp32_tests.cc | 46 parameter.out_strides_[0] = 6; in TEST_F()
47 parameter.out_strides_[1] = 1; in TEST_F()
92 parameter.out_strides_[0] = 2; in TEST_F()
93 parameter.out_strides_[1] = 1; in TEST_F()
140 parameter.out_strides_[0] = 4; in TEST_F()
141 parameter.out_strides_[1] = 1; in TEST_F()
|
| D | transpose_fp32_tests.cc | 139 param->out_strides_[i] = out_strides[i]; in TEST_F()
171 param->out_strides_[i] = out_strides[i]; in TEST_F()
203 param->out_strides_[i] = out_strides[i]; in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | arg_min_max_fp16.c | 117 size_t out_offset = j * param->out_strides_[0] + i; in ArgMinMaxDim0Fp16()
138 size_t out_dim0_offset = i * param->out_strides_[0]; in ArgMinMaxDim1Fp16()
147 size_t out_offset = out_dim0_offset + j + k * param->out_strides_[1]; in ArgMinMaxDim1Fp16()
170 size_t out_dim0_offset = i * param->out_strides_[0]; in ArgMinMaxDim2Fp16()
173 size_t out_dim1_offset = j * param->out_strides_[1] + out_dim0_offset; in ArgMinMaxDim2Fp16()
182 size_t out_offset = out_dim1_offset + k + l * param->out_strides_[2]; in ArgMinMaxDim2Fp16()
206 size_t out_dim0_offset = i * param->out_strides_[0]; in ArgMinMaxDim3Fp16()
209 size_t out_dim1_offset = j * param->out_strides_[1] + out_dim0_offset; in ArgMinMaxDim3Fp16()
212 size_t out_dim2_offset = k * param->out_strides_[2] + out_dim1_offset; in ArgMinMaxDim3Fp16()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | scatter_nd_update_fp32.cc | 79 out_strides_.push_back(1); in ReSize()
82 out_strides_.push_back(out_stride); in ReSize()
84 std::reverse(out_strides_.begin(), out_strides_.end()); in ReSize()
98 tmp_stride += indices_ptr[i * indice_unit_rank + j] * out_strides_.at(j) * unit_size_; in ReSize()
|
| D | roi_pooling_fp32.cc | 69 param_->out_strides_[ndims - 1] = 1; in ReSize()
72 param_->out_strides_[i] = out_shape.at(i + 1) * param_->out_strides_[i + 1]; in ReSize()
|
| D | scatter_nd_fp32.cc | 101 out_strides_.push_back(1); in ReSize()
104 out_strides_.push_back(out_stride); in ReSize()
119 tmp_stride += indices_ptr[i * indice_unit_rank + j] * out_strides_.at(j) * unit_size_; in ReSize()
|
| D | arithmetic_fp32.cc | 149 param_->out_strides_, param_->multiples0_); in ConstTensorBroadCast()
162 param_->out_strides_, param_->multiples1_); in ConstTensorBroadCast()
289 param_->out_strides_[dim] * data_type_len_}; in BroadcastRun()
307 if (break_pos_ > ARITHMETIC_SUPPORT_DIMS_NUM || param_->out_strides_[break_pos_ - 1] == 0) { in BatchScalarCalc()
311 int batch = param_->out_elements_num_ / param_->out_strides_[break_pos_ - 1]; in BatchScalarCalc()
320 int out_stride = param_->out_strides_[break_pos_ - 1] * data_type_len_; in BatchScalarCalc()
329 … static_cast<uint8_t *>(output_ptr_) + out_offset, param_->out_strides_[break_pos_ - 1], true); in BatchScalarCalc()
|
| D | transpose_fp32.cc | 70 param_->out_strides_[param_->num_axes_ - 1] = 1; in ReSize()
76 param_->out_strides_[i] = out_shape.at(i + 1) * param_->out_strides_[i + 1]; in ReSize()
|
| D | arithmetic_compare_fp32.cc | 50 … reinterpret_cast<uint8_t *>(output) + i * param_->out_strides_[dim], dim + 1, out_count, in BroadcastRun()
55 … reinterpret_cast<uint8_t *>(output) + i * param_->out_strides_[dim], dim + 1, out_count, in BroadcastRun()
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/ |
| D | transpose_int8_coder.cc | 42 param_->out_strides_[param_->num_axes_ - 1] = 1; in Prepare()
45 param_->out_strides_[i] = out_shape.at(i + 1) * param_->out_strides_[i + 1]; in Prepare()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | transpose_int8.cc | 66 transpose_param_->out_strides_[transpose_param_->num_axes_ - 1] = 1; in ReSize()
69 transpose_param_->out_strides_[i] = out_shape.at(i + 1) * transpose_param_->out_strides_[i + 1]; in ReSize()
|
| D | scale_int8.cc | 66 … tile_para->in_strides1_, tile_para->out_strides_, tile_para->multiples1_); in InitScaleOffset()
93 … tile_para->in_strides1_, tile_para->out_strides_, tile_para->multiples1_); in InitScaleOffset()
310 … tile_para->in_strides1_, tile_para->out_strides_, tile_para->multiples1_); in Run()
324 … tile_para->in_strides1_, tile_para->out_strides_, tile_para->multiples1_); in Run()
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
| D | transpose_fp32_coder.cc | 43 param_->out_strides_[param_->num_axes_ - 1] = 1; in Resize()
47 param_->out_strides_[i] = out_shape.at(i + 1) * param_->out_strides_[i + 1]; in Resize()
|
| D | arithmetic_fp32_coder.cc | 221 …init_code.CodeArray("out_stride", arithmetic_parameter_->out_strides_, arithmetic_parameter_->ndim… in ConstTensorBroadCast()
236 …init_code.CodeArray("out_stride", arithmetic_parameter_->out_strides_, arithmetic_parameter_->ndim… in ConstTensorBroadCast()
358 …int batch = arithmetic_parameter_->out_elements_num_ / arithmetic_parameter_->out_strides_[break_p… in BatchScalarCalc()
367 int out_stride = arithmetic_parameter_->out_strides_[break_pos_ - 1] * data_type_len_; in BatchScalarCalc()
377 …wrap_uint8(output_ptr_str_), batch_size, arithmetic_parameter_->out_strides_[break_pos_ - 1], true, in BatchScalarCalc()
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/serializers/nnacl_serializer/ |
| D | nnacl_fp32_serializer.cc | 53 … ToString(arithmetic_parameter.out_strides_), ToString(arithmetic_parameter.multiples0_), in CodeStruct()
100 …ToString(tile_parameter.in_strides_), ToString(tile_parameter.out_strides_), tile_parameter.in_dim… in CodeStruct()
107 …ToString(transpose_parameter.out_strides_), transpose_parameter.num_axes_, transpose_parameter.dat… in CodeStruct()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/ |
| D | transpose.h | 35 int out_strides_[MAX_TRANSPOSE_DIM_SIZE]; member
|