add chol

src/backend/tensorflow/graph.jl

@@ -30,6 +30,7 @@ graph(::typeof(mean), x, dim=nothing) = TensorFlow.reduce_mean(x;axis=dim)
 graph(::typeof(svd), x) = svd(x)
 graph(::typeof(size), x, dim) = TensorFlow.size(x,convert(Tensor{Int32}, dim))
 graph(::typeof(size), x) = TensorFlow.size(x)
+graph(::typeof(chol), args...) = TensorFlow.transpose(TensorFlow.cholesky(args...))
 
 for op in (*, .*, .+, .^, log, exp, ceil, floor, sqrt, abs, cos,
            sin, tan, atan, asin, acos, tanh, lgamma, erf, erfc, real, imag, conj,

test/backend/tensorflow.jl

@@ -44,6 +44,9 @@ end
   A = randn(4,5)
   @test tf(@net x -> size(x))(A) == [4,5]
   @test tf(@net (x,y) -> size(x,y))(A,1) == 4
+  A = randn(6,5)
+  A = A'*A
+  @test tf(@net x -> chol(x))(A) ≈ chol(A)
 end
 
 end