saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-29 20:14:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Remove redundant code in ml::write_scores

Browse source
This commit is contained in:
Changming Sun 2019-04-02 17:59:57 -07:00
parent 512cfdd9fe
commit 261a9078a5
1 changed files with 8 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
onnxruntime/core/providers/cpu/ml/ml_common.h
View file

@ -284,10 +284,11 @@ static inline void ComputeSoftmaxZero(std::vector<float>& values) {
}
}
static inline void write_scores(std::vector<float>& scores, POST_EVAL_TRANSFORM post_transform, int64_t write_index, Tensor* Z, int add_second_class) {
template <typename T>
void write_scores(std::vector<T>& scores, POST_EVAL_TRANSFORM post_transform, int64_t write_index, Tensor* Z,
int add_second_class) {
if (post_transform == POST_EVAL_TRANSFORM::PROBIT && scores.size() == 1) {
scores[0] = ComputeProbit(scores[0]);
Z->template MutableData<float>()[write_index] = scores[0];
} else if (scores.size() >= 2) { //multiclass
if (post_transform == POST_EVAL_TRANSFORM::LOGISTIC) {
for (float& score : scores) {
@ -322,10 +323,12 @@ static inline void write_scores(std::vector<float>& scores, POST_EVAL_TRANSFORM
}
}
}
for (float score : scores) {
Z->template MutableData<float>()[write_index] = score;
write_index++;
T* out_p = Z->template MutableData<T>() + write_index;
size_t len;
if (!IAllocator::CalcMemSizeForArray(scores.size(), sizeof(T), &len)) {
ORT_THROW("length overflow");
}
memcpy(out_p, scores.data(), len);
}
} // namespace ml