Searched refs:input_unit_ (Results 1 – 14 of 14) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | convolution_winograd_fp32.cc | 36 … input_unit_, kernel_unit_, conv_param_->input_channel_, conv_param_->output_channel_, in WinogradFilterTransform()
43 …thread_count_ * tile_num_ * input_unit_ * input_unit_ * conv_param_->input_channel_ * sizeof(float… in InitTmpBuffer()
52 …ctx_->allocator->Malloc(thread_count_ * tile_num_ * input_unit_ * input_unit_ * oc8 * sizeof(float… in InitTmpBuffer()
59 ctx_->allocator->Malloc(thread_count_ * C4NUM * input_unit_ * input_unit_ * sizeof(float))); in InitTmpBuffer()
80 in_func_ = GetInputTransFunc(input_unit_); in ConfigInputOutput()
85 out_func_ = GetOutputTransFunc(input_unit_, output_unit_, conv_param_->act_type_); in ConfigInputOutput()
103 input_unit_ = output_unit_ + kernel_unit_ - 1; in Init()
104 conv_param_->input_unit_ = input_unit_; in Init()
112 input_unit_ * input_unit_ * in_channel * UP_ROUND(out_channel, oc_block_) * sizeof(float); in Init()
205 input_unit_ * input_unit_ * in_channel * UP_ROUND(out_channel, oc_block_) * sizeof(float); in MallocWeightBiasData()
[all …]
|
| D | convolution_winograd_fp32.h | 67 int input_unit_{0};
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
| D | convolution_winograd_fp32_coder.cc | 81 …ightTransform(weight_data, trans_weight_, matrix_g, matrix_gt, oc_block, input_unit_, kernel_unit_, in WinogradFilterTransform()
89 …tile_buffer_size_ = thread_num_ * tile_num * input_unit_ * input_unit_ * conv_param_->input_channe… in InitTmpBuffer()
91 gemm_out_size_ = thread_num_ * tile_num * input_unit_ * input_unit_ * oc8 * C8NUM * sizeof(float); in InitTmpBuffer()
93 tmp_data_size_ = thread_num_ * C4NUM * input_unit_ * input_unit_ * sizeof(float); in InitTmpBuffer()
113 input_unit_ = output_unit_ + kernel_unit_ - 1; in Prepare()
114 conv_param_->input_unit_ = input_unit_; in Prepare()
133 int trans_matrix_data_size = input_unit_ * input_unit_ * in_channel * oc_block_num * oc_block; in InitWeightBias()
146 if (input_unit_ == DIMENSION_8D) { in InitWeightBias()
175 in_func_ = GetInputTransFunc(input_unit_); in ConfigInputOutput()
177 out_func_ = GetOutputTransFunc(input_unit_, output_unit_, conv_param_->act_type_); in ConfigInputOutput()
[all …]
|
| D | convolution_winograd_fp32_coder.h | 58 int input_unit_{0};
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | convolution_winograd_fp16.cc | 31 … oc_block, input_unit_, kernel_unit_, conv_param_->input_channel_, in WinogradFilterTransformFp16()
43 …auto trans_matrix_data_size = input_unit_ * input_unit_ * in_channel * oc_block_num * col_tile_ * … in MallocWeightBiasData()
60 if (input_unit_ == 8) { in MallocWeightBiasData()
91 …thread_count_ * row_tile_ * input_unit_ * input_unit_ * conv_param_->input_channel_ * sizeof(float… in InitTmpBuffer()
99 …thread_count_ * row_tile_ * input_unit_ * input_unit_ * UP_ROUND(channel_out, C8NUM) * sizeof(floa… in InitTmpBuffer()
106 ctx_->allocator->Malloc(thread_count_ * C8NUM * input_unit_ * input_unit_ * sizeof(float16_t))); in InitTmpBuffer()
127 in_func_ = GetInputTransFp16Func(input_unit_); in ConfigInputOutput()
132 out_func_ = GetOutputTransFp16Func(input_unit_, output_unit_, conv_param_->act_type_); in ConfigInputOutput()
156 …auto trans_matrix_data_size = input_unit_ * input_unit_ * in_channel * oc_block_num * col_tile_ * … in Init()
160 input_unit_ = output_unit_ + kernel_unit_ - 1; in Init()
[all …]
|
| D | convolution_winograd_fp16.h | 71 int input_unit_ = 0; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | conv_winograd_fp32.c | 43 int input_unit_square = conv_param->input_unit_ * conv_param->input_unit_; in ConvWinogardFp32()
44 if (input_unit_square < conv_param->input_unit_) { in ConvWinogardFp32()
|
| D | winograd_transform.c | 24 int input_unit = conv_param->input_unit_; in WinogradInputTransform()
109 int input_unit = conv_param->input_unit_; in WinogradOutputNHWCTransform()
169 int input_unit = conv_param->input_unit_; in WinogradOutputNC4HW4Transform()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | winograd_transform_fp16.c | 28 int input_unit = conv_param->input_unit_; in WinogradInputTransformFp16()
136 int input_unit = conv_param->input_unit_; in WinogradOutputNHWCTransformFp16()
176 int input_unit = conv_param->input_unit_; in WinogradOutputNC8HW8TransformFp16()
|
| D | conv_fp16.c | 207 int input_unit_square = conv_param->input_unit_ * conv_param->input_unit_; in ConvWinogardFp16()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/ |
| D | minimal_filtering_generator.h | 53 … int oc_block, int input_unit_, int kernel_unit_, int channel, int batch, bool pack);
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/ |
| D | conv_parameter.h | 50 int input_unit_; member
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/serializers/nnacl_serializer/ |
| D | nnacl_fp32_serializer.cc | 70 …conv_parameter.output_channel_, "thread_num", conv_parameter.input_unit_, conv_parameter.output_un… in CodeStruct()
|
| D | nnacl_int8_serializer.cc | 60 …conv_parameter.input_unit_, conv_parameter.output_unit_, conv_parameter.pad_mode_, conv_parameter.… in CodeStruct()
|