summaryrefslogtreecommitdiff
path: root/ot/lp/EMD.h
diff options
context:
space:
mode:
authorRémi Flamary <remi.flamary@gmail.com>2019-12-03 15:20:16 +0100
committerRémi Flamary <remi.flamary@gmail.com>2019-12-03 15:20:16 +0100
commita4afee871d8e9d5db68228d1ed5bf4853eedc294 (patch)
tree9aed5291d5e472085b0eca3d0453865684bd443d /ot/lp/EMD.h
parentc439e3efb920086154c741b41f65d99165e875d8 (diff)
first implemntation sparse loss
Diffstat (limited to 'ot/lp/EMD.h')
-rw-r--r--ot/lp/EMD.h5
1 files changed, 5 insertions, 0 deletions
diff --git a/ot/lp/EMD.h b/ot/lp/EMD.h
index 9896091..fc94211 100644
--- a/ot/lp/EMD.h
+++ b/ot/lp/EMD.h
@@ -36,4 +36,9 @@ int EMD_wrap_return_sparse(int n1, int n2, double *X, double *Y, double *D,
long *iG, long *jG, double *G, long * nG,
double* alpha, double* beta, double *cost, int maxIter);
+int EMD_wrap_all_sparse(int n1, int n2, double *X, double *Y,
+ long *iD, long *jD, double *D, long nD,
+ long *iG, long *jG, double *G, long * nG,
+ double* alpha, double* beta, double *cost, int maxIter);
+
#endif