Searched refs:reduce_param (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/ops/populate/v0/ |
| D | reduce_populate_v0.cc | 32 auto *reduce_param = reinterpret_cast<ReduceParameter *>(malloc(sizeof(ReduceParameter))); in PopulateReduceParameter() local
33 if (reduce_param == nullptr) { in PopulateReduceParameter()
37 memset(reduce_param, 0, sizeof(ReduceParameter)); in PopulateReduceParameter()
38 reduce_param->op_parameter_.type_ = schema::PrimitiveType_ReduceFusion; in PopulateReduceParameter()
40 reduce_param->keep_dims_ = static_cast<bool>(reduce_prim->keepDims()); in PopulateReduceParameter()
41 reduce_param->reduce_to_end_ = reduce_prim->reduceToEnd(); in PopulateReduceParameter()
42 reduce_param->coeff = reduce_prim->coeff(); in PopulateReduceParameter()
46 free(reduce_param); in PopulateReduceParameter()
51 free(reduce_param); in PopulateReduceParameter()
54 reduce_param->num_axes_ = static_cast<int>(axisVector->size()); in PopulateReduceParameter()
[all …]
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/caffe/ |
| D | caffe_reduce_parser.cc | 32 const caffe::ReductionParameter &reduce_param = proto.reduction_param(); in Parse() local
33 if (reduce_param.has_operation()) { in Parse()
34 if (reduce_param.operation() == caffe::ReductionParameter_ReductionOp_MEAN) { in Parse()
36 } else if (reduce_param.operation() == caffe::ReductionParameter_ReductionOp_SUM) { in Parse()
38 } else if (reduce_param.operation() == caffe::ReductionParameter_ReductionOp_SUMSQ) { in Parse()
40 } else if (reduce_param.operation() == caffe::ReductionParameter_ReductionOp_ASUM) { in Parse()
43 MS_LOG(ERROR) << "nsupported reduce mode: " << reduce_param.operation(); in Parse()
51 if (reduce_param.has_axis()) { in Parse()
52 axes = std::vector<int>(1, reduce_param.axis()); in Parse()
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/base/ |
| D | reduce_base_coder.cc | 76 auto reduce_param = reinterpret_cast<ReduceParameter *>(parameter_); in Init() local
77 if (reduce_param == nullptr) { in Init()
86 num_axes_ = reduce_param->num_axes_; in Init()
87 …MS_CHECK_RET_CODE(memcpy_s(axes_, sizeof(axes_), reduce_param->axes_, sizeof(reduce_param->axes_)), in Init()
90 mode_ = reduce_param->mode_; in Init()
91 …MS_CHECK_RET_CODE(memcpy_s(axes_, sizeof(axes_), reduce_param->axes_, sizeof(reduce_param->axes_)), in Init()
93 reduce_to_end_ = reduce_param->reduce_to_end_; in Init()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | reduce_base.cc | 99 auto reduce_param = reinterpret_cast<ReduceParameter *>(op_parameter_); in Init() local
100 if (reduce_param == nullptr) { in Init()
112 if (reduce_param->keep_dims_) { in Init()
125 num_axes_ = reduce_param->num_axes_; in Init()
126 memcpy(axes_, reduce_param->axes_, sizeof(reduce_param->axes_)); in Init()
129 mode_ = reduce_param->mode_; in Init()
130 reduce_to_end_ = reduce_param->reduce_to_end_; in Init()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/ |
| D | reduce.cc | 55 auto reduce_param = reinterpret_cast<ReduceParameter *>(op_parameter_); in GenC4Mask() local
61 float init_float = reduce_type2init.find(reduce_param->mode_)->second; in GenC4Mask()
139 auto reduce_param = reinterpret_cast<ReduceParameter *>(op_parameter_); in CheckSpecs() local
140 CHECK_NULL_RETURN(reduce_param); in CheckSpecs()
141 if (GetReduceTypeStr(reduce_param->mode_).empty()) { in CheckSpecs()
142 MS_LOG(WARNING) << "not supported reduce type:" << reduce_param->mode_; in CheckSpecs()
156 if ((c_reduce_ || wc_reduce_) && !reduce_param->keep_dims_) { in CheckSpecs()
164 auto reduce_param = reinterpret_cast<ReduceParameter *>(op_parameter_); in Prepare() local
165 if (reduce_param == nullptr) { in Prepare()
187 kernel_name += GetReduceTypeStr(reduce_param->mode_); in Prepare()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | reduce_fp16.cc | 150 auto reduce_param = reinterpret_cast<ReduceParameter *>(opParameter); in CpuReduceFp16KernelCreator() local
151 if (reduce_param->mode_ != ReduceMode_ReduceMean && reduce_param->mode_ != ReduceMode_ReduceMax && in CpuReduceFp16KernelCreator()
152 reduce_param->mode_ != ReduceMode_ReduceSum) { in CpuReduceFp16KernelCreator()
153 MS_LOG(ERROR) << "Reduce unsupported reduce mode: " << reduce_param->mode_; in CpuReduceFp16KernelCreator()
|