add more basic math ops to tf backend

2017-05-09 14:39:10 +02:00 · 2017-05-09 14:39:10 +02:00 · d5b073db74
commit d5b073db74
parent eda5c2c776
1 changed files with 2 additions and 2 deletions
--- a/src/backend/tensorflow/graph.jl
+++ b/src/backend/tensorflow/graph.jl
@ -29,8 +29,8 @@ graph(::typeof(all), x, dim) = TensorFlow.reduce_all(x;axis=dim)
 graph(::typeof(any), x, dim) = TensorFlow.reduce_any(x;axis=dim)
 graph(::typeof(mean), x, dim) = TensorFlow.reduce_mean(x;axis=dim)

-
-for op in (tanh, *, .*, .+, .^)
+for op in (*, .*, .+, .^, log, exp, ceil, floor, sqrt, abs, cos, 
+           sin, tan, atan, asin, acos, tanh, lgamma, erf, erfc, real, imag, conj)
  @eval graph(::typeof($op), args...) = $op(args...)
 end