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

move op

Browse Source
This commit is contained in:
Mike J Innes 2016-11-13 18:16:58 +00:00
parent e4a6ca5f9e
commit 6ac4dd8429
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -35,13 +35,6 @@ graph(::DataFlow.Group, xs...) = (xs...,)
graph(params::Associative, c::Conv2D, x) = graph(params::Associative, c::Conv2D, x) =
nn.conv2d(x, graph(params, c.filter), [1,c.stride...,1], "VALID") nn.conv2d(x, graph(params, c.filter), [1,c.stride...,1], "VALID")
type Op
f
shape
end
Op(f) = Op(f, (d...) -> nothing)
graph(op::Op, xs...) = op.f(xs...) graph(op::Op, xs...) = op.f(xs...)
Flux.shape(op::Op, d...) = op.shape(d...) Flux.shape(op::Op, d...) = op.shape(d...)

7
src/backend/tensorflow/tensorflow.jl
View File

@ -5,6 +5,13 @@ import Flux: accuracy, spliceinputs, detuple
export tf export tf
type Op
f
shape
end
Op(f) = Op(f, (d...) -> nothing)
include("graph.jl") include("graph.jl")
include("model.jl") include("model.jl")
include("recurrent.jl") include("recurrent.jl")