Searched refs:one_hot_param (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/ops/populate/v0/ |
| D | one_hot_populate_v0.cc | 32 auto *one_hot_param = reinterpret_cast<OneHotParameter *>(malloc(sizeof(OneHotParameter))); in PopulateOneHotParameter() local
33 if (one_hot_param == nullptr) { in PopulateOneHotParameter()
37 memset(one_hot_param, 0, sizeof(OneHotParameter)); in PopulateOneHotParameter()
38 one_hot_param->op_parameter_.type_ = schema::PrimitiveType_OneHot; in PopulateOneHotParameter()
39 one_hot_param->axis_ = one_hot_prim->axis(); in PopulateOneHotParameter()
40 return reinterpret_cast<OpParameter *>(one_hot_param); in PopulateOneHotParameter()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | one_hot_fp32.c | 21 const OneHotParameter *one_hot_param, const int tid, const int thread_num) { in OneHotToFp32() argument
22 if (indices == NULL || one_hot_param == NULL || output == NULL) { in OneHotToFp32()
29 int outer_size = one_hot_param->outer_size_; in OneHotToFp32()
30 int inner_size = one_hot_param->inner_size_; in OneHotToFp32()
31 int depth = one_hot_param->depth_; in OneHotToFp32()
40 if (one_hot_param->support_neg_index_ && index < 0) { in OneHotToFp32()
|
| D | one_hot_fp32.h | 30 const OneHotParameter *one_hot_param, const int tid, const int thread_num);
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | one_hot_fp16.c | 20 const OneHotParameter *one_hot_param, const int tid, const int thread_num) { in OneHotToFp16() argument
21 if (indices == NULL || one_hot_param == NULL || output == NULL) { in OneHotToFp16()
28 int outer_size = one_hot_param->outer_size_; in OneHotToFp16()
29 int inner_size = one_hot_param->inner_size_; in OneHotToFp16()
30 int depth = one_hot_param->depth_; in OneHotToFp16()
39 if (one_hot_param->support_neg_index_ && index < 0) { in OneHotToFp16()
|
| D | one_hot_fp16.h | 30 const OneHotParameter *one_hot_param, const int tid, const int thread_num);
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | one_hot_base.cc | 118 auto one_hot_param = reinterpret_cast<OneHotParameter *>(op_parameter_); in OneHotImpl() local
121 …tToFp32(indices_data, on_value_, off_value_, reinterpret_cast<float *>(output_data), one_hot_param, in OneHotImpl()
127 one_hot_param, task_id, thread_num_); in OneHotImpl()
137 auto one_hot_param = reinterpret_cast<OneHotParameter *>(op_parameter_); in InitParamsAndOnOffValue() local
138 if (one_hot_param == nullptr) { in InitParamsAndOnOffValue()
152 one_hot_param->depth_ = *depth; in InitParamsAndOnOffValue()
156 one_hot_param->support_neg_index_ = false; in InitParamsAndOnOffValue()
164 one_hot_param->support_neg_index_ = true; in InitParamsAndOnOffValue()
172 one_hot_param->outer_size_ = outer_size_; in InitParamsAndOnOffValue()
173 one_hot_param->inner_size_ = inner_size_; in InitParamsAndOnOffValue()
|