diff options
author | Håvard Pettersen <havardpe@oath.com> | 2021-04-12 13:39:47 +0000 |
---|---|---|
committer | Håvard Pettersen <havardpe@oath.com> | 2021-04-14 08:13:22 +0000 |
commit | fd3d1ebd87ee376e0995681f713808eb67a0142a (patch) | |
tree | 6616ad3de89ad1384dfb42f0097b2306c0e3f2ae /eval | |
parent | 806aae9a0c3c1e962539990d53792d7427902f30 (diff) |
add pending stuff (erf/median)
Diffstat (limited to 'eval')
-rw-r--r-- | eval/src/apps/tensor_conformance/generate.cpp | 6 |
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); } |