saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-24 02:47:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Optimised kernel_dot() in SVM op (#3135)

Browse source
This commit is contained in:
Prabhat 2020-03-04 16:30:40 +00:00 committed by GitHub
parent 9d874c1225
commit a2eeb126b9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 24 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

43
onnxruntime/core/providers/cpu/ml/svmclassifier.h
View file

@ -33,25 +33,30 @@ class SVMCommon {
double sum = 0;
const T* pA = A + a;
const float* pB = B.data() + b;
if (k == KERNEL::POLY) {
for (int64_t i = len; i > 0; --i, ++pA, ++pB)
sum += *pA * *pB;
sum = gamma_ * sum + coef0_;
sum = std::pow(sum, degree_);
} else if (k == KERNEL::SIGMOID) {
for (int64_t i = len; i > 0; --i, ++pA, ++pB)
sum += *pA * *pB;
sum = gamma_ * sum + coef0_;
sum = std::tanh(sum);
} else if (k == KERNEL::RBF) {
for (int64_t i = len; i > 0; --i, ++pA, ++pB) {
double val = *pA - *pB;
sum += val * val;
}
sum = std::exp(-gamma_ * sum);
} else if (k == KERNEL::LINEAR) {
for (int64_t i = len; i > 0; --i, ++pA, ++pB)
sum += *pA * *pB;
switch(k) {
case KERNEL::POLY:
for (int64_t i = len; i > 0; --i)
sum += *pA++ * *pB++;
sum = gamma_ * sum + coef0_;
sum = std::pow(sum, degree_);
break;
case KERNEL::SIGMOID:
for (int64_t i = len; i > 0; --i)
sum += *pA++ * *pB++;
sum = gamma_ * sum + coef0_;
sum = std::tanh(sum);
break;
case KERNEL::RBF:
for (int64_t i = len; i > 0; --i) {
double val = *pA++ - *pB++;
sum += val * val;
}
sum = std::exp(-gamma_ * sum);
break;
case KERNEL::LINEAR:
for (int64_t i = len; i > 0; --i)
sum += *pA++ * *pB++;
break;
}
return (float)sum;
}