summaryrefslogtreecommitdiffstats
path: root/eval
diff options
context:
space:
mode:
authorHåvard Pettersen <havardpe@oath.com>2021-04-12 13:39:47 +0000
committerHåvard Pettersen <havardpe@oath.com>2021-04-14 08:13:22 +0000
commitfd3d1ebd87ee376e0995681f713808eb67a0142a (patch)
tree6616ad3de89ad1384dfb42f0097b2306c0e3f2ae /eval
parent806aae9a0c3c1e962539990d53792d7427902f30 (diff)
add pending stuff (erf/median)
Diffstat (limited to 'eval')
-rw-r--r--eval/src/apps/tensor_conformance/generate.cpp6
1 files changed, 2 insertions, 4 deletions
diff --git a/eval/src/apps/tensor_conformance/generate.cpp b/eval/src/apps/tensor_conformance/generate.cpp
index 8a81e53b9ec..ce7cdb239b2 100644
--- a/eval/src/apps/tensor_conformance/generate.cpp
+++ b/eval/src/apps/tensor_conformance/generate.cpp
@@ -48,8 +48,7 @@ void generate_tensor_reduce(TestBuilder &dst) {
generate_reduce(Aggr::PROD, SigmoidF(N()), dst);
generate_reduce(Aggr::SUM, N(), dst);
generate_reduce(Aggr::MAX, N(), dst);
- // add MEDIAN cases when supported in Java
- // generate_reduce(Aggr::MEDIAN, N(), dst);
+ generate_reduce(Aggr::MEDIAN, N(), dst);
generate_reduce(Aggr::MIN, N(), dst);
}
@@ -103,8 +102,7 @@ void generate_tensor_map(TestBuilder &dst) {
generate_op1_map("relu(a)", Sub2(Div16(N())), dst);
generate_op1_map("sigmoid(a)", Sub2(Div16(N())), dst);
generate_op1_map("elu(a)", Sub2(Div16(N())), dst);
- // TODO(havardpe): add erf when supported by Java
- // generate_op1_map("erf(a)", operation::Erf::f, Sub2(Div16(N())), dst);
+ generate_op1_map("erf(a)", Sub2(Div16(N())), dst);
generate_op1_map("a in [1,5,7,13,42]", N(), dst);
generate_map_expr("map(a,f(a)((a+1)*2))", Div16(N()), dst);
}