Searched refs:weight_origin (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | convolution_1x1_fp16.cc | 118 void *weight_origin = (op_parameter_->is_train_session_) ? weight_tensor->data() : origin_weight_; in PackWeight() local
119 MS_ASSERT(weight_origin != nullptr); in PackWeight()
122 …ColMajor2Row8MajorFp16(weight_origin, reinterpret_cast<float16_t *>(packed_weight_), input_channel… in PackWeight()
125 RowMajor2Col16MajorFp16Opt(static_cast<const float16_t *>(weight_origin), in PackWeight()
129 …ColMajor2Row8MajorFp16(weight_origin, reinterpret_cast<float16_t *>(packed_weight_), input_channel… in PackWeight()
|
| D | convolution_fp16.cc | 35 void *weight_origin = (op_parameter_->is_train_session_) ? filter_tensor->data() : origin_weight_; in PackWeight() local
36 MS_ASSERT(weight_origin != nullptr); in PackWeight()
37 RowMajor2Col8MajorFp16(weight_origin, reinterpret_cast<float16_t *>(packed_weight_), out_channel, in PackWeight()
|
| D | convolution_winograd_fp16.cc | 83 void *weight_origin = (op_parameter_->is_train_session_) ? weight_tensor->data() : origin_weight_; in PackWeight() local
84 MS_ASSERT(weight_origin != nullptr); in PackWeight()
85 …WinogradFilterTransformFp16(reinterpret_cast<float16_t *>(weight_origin), matrix_g_, matrix_gt_, c… in PackWeight()
|