experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Go: Update generated wrapper functions for TensorFlow ops.

Browse Source 
PiperOrigin-RevId: 157573997
This commit is contained in:
A. Unique TensorFlower 2017-05-31 04:53:31 -07:00 committed by TensorFlower Gardener
parent 7f96742174
commit eeb0b40670
1 changed files with 11 additions and 2 deletions

13
tensorflow/go/op/wrappers.go
View File

@ -7948,10 +7948,19 @@ func WriteFile(scope *Scope, filename tf.Output, contents tf.Output) (o *tf.Oper
// Computes the Cholesky decomposition of one or more square matrices.
//
// The input is a tensor of shape `[..., M, M]` whose inner-most 2 dimensions
// form square matrices, with the same constraints as the single matrix Cholesky
// decomposition above. The output is a tensor of the same shape as the input
// form square matrices.
//
// The input has to be symmetric and positive definite. Only the lower-triangular
// part of the input will be used for this operation. The upper-triangular part
// will not be read.
//
// The output is a tensor of the same shape as the input
// containing the Cholesky decompositions for all input submatrices `[..., :, :]`.
//
// **Note**: The gradient computation on GPU is faster for large matrices but
// not for large batch dimensions when the submatrices are small. In this
// case it might be faster to use the CPU.
//
// Arguments:
// input: Shape is `[..., M, M]`.
//