Searched refs:deep_4_ (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/lite/micro/coder/wrapper/int8/ |
| D | conv1x1_run_int8_wrapper.c | 42 …n = args->packed_input_ + task_id * args->thread_stride_hw_ * C4NUM * args->matmul_param_->deep_4_; in OcOptPre()
73 …nv1x1Int8Opt(args->packed_input_, args->packed_weight_ + cur_stride * args->matmul_param_->deep_4_, in RunArm64OptOc()
75 … args->matmul_param_->row_, cur_oc, args->matmul_param_->deep_4_, cur_left_shift, cur_right_shift, in RunArm64OptOc()
121 …n = args->packed_input_ + task_id * args->thread_stride_hw_ * C4NUM * args->matmul_param_->deep_4_; in RunArm64OptHw()
132 … args->matmul_param_->col_, args->matmul_param_->deep_4_, args->left_shift_, args->right_shift_, in RunArm64OptHw()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | convolution_1x1_int8.cc | 343 matmul_param_->deep_4_ = UP_ROUND(matmul_param_->deep_, C4NUM); in InitParam()
458 …nt8_t *hw_packed_in = packed_input_ + task_id * thread_stride_hw_ * C4NUM * matmul_param_->deep_4_; in RunArm64OptHw()
469 … matmul_param_->col_, matmul_param_->deep_4_, left_shift_, right_shift_, multiplier_, conv_param_, in RunArm64OptHw()
491 …Conv1x1Int8Opt(packed_input_, packed_weight_ + cur_stride * matmul_param_->deep_4_, output_ptr_ + … in RunArm64OptOc()
493 …matmul_param_->deep_4_, cur_left_shift, cur_right_shift, cur_multiplier, conv_param_, matmul_func_, in RunArm64OptOc()
535 …nt8_t *hw_packed_in = packed_input_ + task_id * thread_stride_hw_ * C4NUM * matmul_param_->deep_4_; in OcOptPre()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/ |
| D | matmul_parameter.h | 55 int deep_4_; member
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/ |
| D | conv2d_1x1_int8_coder.cc | 242 matmul_param_->deep_4_ = UP_ROUND(matmul_param_->deep_, C4NUM); in InitParam()
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/serializers/nnacl_serializer/ |
| D | nnacl_fp32_serializer.cc | 80 …mat_mul_parameter.col_align_, mat_mul_parameter.deep_, mat_mul_parameter.deep_4_, mat_mul_paramete… in CodeStruct()
|
| D | nnacl_int8_serializer.cc | 69 matmul_parameter.deep_, matmul_parameter.deep_4_, matmul_parameter.deep_16_, in CodeStruct()
|