Home
last modified time | relevance | path

Searched refs:row_align_ (Results 1 – 13 of 13) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Ddeconvolution_fp32.cc106 matmul_param_->row_align_ = UP_ROUND(matmul_param_->row_, row_tile_); in InitParam()
139 …uffer = tmp_buffer_ + task_id * thread_stride_ * C8NUM * kernel_plane_ * matmul_param_->row_align_; in DoDeconv()
144 …tmp_buffer, matmul_param_->deep_, matmul_param_->row_align_, oc * C8NUM * kernel_plane_, kernel_pl… in DoDeconv()
149 tmp_buffer, matmul_param_->deep_, matmul_param_->row_align_, oc * C8NUM * kernel_plane_); in DoDeconv()
154 …tmp_buffer, nullptr, ActType_No, matmul_param_->deep_, matmul_param_->row_align_, oc * C8NUM * ker… in DoDeconv()
231 ctx_->allocator->Malloc(matmul_param_->row_align_ * matmul_param_->col_8_ * sizeof(float))); in InitRunBuf()
238 ctx_->allocator->Malloc(matmul_param_->row_align_ * matmul_param_->deep_ * sizeof(float))); in InitRunBuf()
Dmatmul_fp32_base.cc66 params_->row_align_ = UP_ROUND(params_->row_, row_tile_); in ResizeParameter()
179 float *dst = a_pack_ptr_ + i * params_->deep_ * params_->row_align_; in InitMatrixA()
326 params_->row_align_ = UP_ROUND(params_->row_, row_tile_); in init_global_variable()
334 matrix_a_pack_size_ = params_->batch * params_->row_align_ * params_->deep_; in Init()
379 matrix_a_pack_size_ = params_->batch * params_->row_align_ * params_->deep_; in ReSize()
474 batch_a_ptr_ = a_pack_ptr_ + i * params_->row_align_ * params_->deep_; in Run()
Dconvolution_1x1_fp32.cc60 matmul_param_->row_align_ = UP_ROUND(matmul_param_->row_, row_tile_); in InitConv1x1MatmulParam()
230 : (matmul_param_->row_align_ * matmul_param_->deep_); in Run()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Dmatmul_base_fp16.cc124 params_->row_align_ = 1; in ResizeParameter()
131 params_->row_align_ = UP_ROUND(params_->row_, row_tile_); in ResizeParameter()
138 …ms_context_->allocator->Malloc(params_->batch * params_->row_align_ * params_->deep_ * sizeof(floa… in InitBufferA()
143 memset(a_pack_ptr_, 0, params_->batch * params_->row_align_ * params_->deep_ * sizeof(float16_t)); in InitBufferA()
178 float16_t *dst = a_pack_ptr_ + i * params_->deep_ * params_->row_align_; in InitMatrixA()
343 batch_a_ptr_ = a_pack_ptr_ + i * params_->row_align_ * params_->deep_; in Run()
Dconvolution_1x1_fp16.cc34 matmul_param_->row_align_ = UP_ROUND(matmul_param_->row_, row_tile_); in InitMatmulParam()
287 ctx_->allocator->Malloc(matmul_param_->row_align_ * matmul_param_->deep_ * sizeof(float16_t))); in Run()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dmatmul_base_int8.cc178 param_->row_align_ = UP_ROUND(param_->row_, row_tile_); in ResizeParameter()
235 …pack_a_ptr_ = reinterpret_cast<int8_t *>(malloc(param_->row_align_ * param_->deep_16_ * sizeof(int… in InitTmpBuffer()
246 input_sums_ = reinterpret_cast<int *>(malloc(param_->row_align_ * sizeof(int))); in InitTmpBuffer()
257 memset(pack_a_ptr_, 0, param_->row_align_ * param_->deep_16_ * sizeof(int8_t)); in InitTmpBuffer()
259 memset(input_sums_, 0, param_->row_align_ * sizeof(int)); in InitTmpBuffer()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/
Dmatmul_fp32_base_coder.cc74 params_->row_align_ = vec_matmul_ ? 1 : UP_ROUND(params_->row_, row_tile_); in ResizeParameter()
82 …a_pack_ptr_size_ = static_cast<size_t>(params_->batch * params_->row_align_ * params_->deep_ * siz… in InitBufferA()
243 …code << "\t\tfloat *batch_a_ptr = " << a_pack_str << " + i * " << params_->row_align_ * params_->d… in DoCode()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/
Dmatmul_cpu_kernel.cc50 const size_t size = param_.batch * param_.row_align_ * param_.deep_; in InitMatrixA()
67 float *dst = a_pack_ptr_ + i * param_.row_align_ * param_.deep_; in InitMatrixA()
149 param_.row_align_ = vec_matmul_ ? 1 : UP_ROUND(param_.row_, row_tile_); in InitArmKernel()
225 batch_a_ptr_ = a_pack_ptr_ + i * param_.row_align_ * param_.deep_; in ParallelRun()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Dmatmul_base_int8_coder.cc34 a_pack_ptr_size_ = param_->row_align_ * param_->deep_16_ * sizeof(int8_t); in InitTmpBuffer()
44 input_sums_size_ = static_cast<size_t>(param_->row_align_ * sizeof(int)); in InitTmpBuffer()
61 param_->row_align_ = UP_ROUND(param_->row_, row_tile_); in ResizeParameter()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dmatmul_parameter.h50 int row_align_; member
/third_party/mindspore/mindspore/lite/micro/coder/wrapper/fp32/
Dmatmul_fp32_wrapper.c25 float *dst = dst_ptr + i * params_->deep_ * params_->row_align_; in InitMatrixA()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/serializers/nnacl_serializer/
Dnnacl_fp32_serializer.cc79 …mat_mul_parameter.row_16_, mat_mul_parameter.row_align_, mat_mul_parameter.col_4_, mat_mul_paramet… in CodeStruct()
Dnnacl_int8_serializer.cc67 … matmul_parameter.row_12_, matmul_parameter.row_16_, matmul_parameter.row_align_, in CodeStruct()