experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
STT-tensorflow/tensorflow/compiler/xla/client/lib
History
Thomas Joerg e60c1ba960 [XLA] Improve numerical stability of Logistic. 
PiperOrigin-RevId: 315320526
Change-Id: Iedfd22d0fb657cb31dda537786ce001f1dab168b
2020-06-08 12:01:07 -07:00
..
arithmetic_test.cc
[XLA] Implement stable argmin and argmax
2020-04-18 00:09:30 -07:00
arithmetic.cc
[XLA] Implement stable argmin and argmax
2020-04-18 00:09:30 -07:00
arithmetic.h
[XLA] Implement stable argmin and argmax
2020-04-18 00:09:30 -07:00
BUILD
[TF/XLA] Change XLA:GPU and XLA:CPU argmax/argmin implementation to return the smallest index corresponding to the largest/smallest number
2020-04-09 14:01:22 -07:00
comparators_test.cc
…
comparators.cc
Change CreateScalarComparisonComputation to have an option to have a vector of XlaOpGenerator as input.
2020-03-17 11:28:25 -07:00
comparators.h
Change CreateScalarComparisonComputation to have an option to have a vector of XlaOpGenerator as input.
2020-03-17 11:28:25 -07:00
constants_test.cc
…
constants.cc
…
constants.h
…
conv_grad_size_util.cc
…
conv_grad_size_util.h
…
logdet_test.cc
…
logdet.cc
…
logdet.h
…
loops.cc
…
loops.h
…
math_test.cc
Add F64 Sqrt test.
2020-05-16 20:03:50 -07:00
math.cc
[XLA] Improve numerical stability of Logistic.
2020-06-08 12:01:07 -07:00
math.h
Add XLA Op registrations for IgammaGradA and RandomGammaGrad.
2020-02-18 16:26:46 -08:00
matrix_test.cc
…
matrix.cc
…
matrix.h
…
pooling_test.cc
…
pooling.cc
…
pooling.h
…
prng.cc
Modify the XLA Uniform sampler to use cast instead of bitcasts. We didn't strictly need a bitcast because we are ignoring the exponent bits anyway.
2020-03-31 15:22:55 -07:00
prng.h
Introduce new RngBitGenerator HLO
2020-02-06 05:06:44 -08:00
qr_test.cc
…
qr.cc
…
qr.h
…
quantize_test.cc
…
quantize.h
…
self_adjoint_eig_test.cc
…
self_adjoint_eig.cc
…
self_adjoint_eig.h
…
slicing_test.cc
…
slicing.cc
Add utility API to implement strided dynamic slice.
2020-03-15 11:05:11 -07:00
slicing.h
Add utility API to implement strided dynamic slice.
2020-03-15 11:05:11 -07:00
sorting_test.cc
Support TopK op to split sorting into smaller sorts and a big final sort.
2020-02-12 12:39:22 -08:00
sorting.cc
Support TopK op to split sorting into smaller sorts and a big final sort.
2020-02-12 12:39:22 -08:00
sorting.h
Support TopK op to split sorting into smaller sorts and a big final sort.
2020-02-12 12:39:22 -08:00
svd_test.cc
…
svd.cc
…
svd.h
…
testing.cc
…
testing.h
…
tridiagonal_test.cc
Unlock XLA non pivoting tridiagonal solver for all back-ends.
2020-03-23 17:23:42 -07:00
tridiagonal.cc
Unlock XLA non pivoting tridiagonal solver for all back-ends.
2020-03-23 17:23:42 -07:00
tridiagonal.h
…