TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add .^ and reduction ops for tf backend

Browse Source
This commit is contained in:
Ali Hamdi 2017-05-09 01:29:15 +02:00
parent 7a85eff370
commit eda5c2c776
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/backend/tensorflow/graph.jl
View File

@ -21,8 +21,16 @@ graph(::typeof(relu), x) = nn.relu(x)
graph(::typeof(σ), x) = nn.sigmoid(x)
graph(::typeof(hcat), xs...) = concat(1, xs)
graph(::typeof(seq), xs, n) = TensorFlow.unpack(xs, num = n, axis = 1)
graph(::typeof(sum), x, dim) = TensorFlow.reduce_sum(x;axis=dim)
graph(::typeof(prod), x, dim) = TensorFlow.reduce_prod(x;axis=dim)
graph(::typeof(min), x, dim) = TensorFlow.reduce_min(x;axis=dim)
graph(::typeof(max), x, dim) = TensorFlow.reduce_max(x;axis=dim)
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 (tanh, *, .*, .+, .^)
@eval graph(::typeof($op), args...) = $op(args...)
end