experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
rei/fork-r2.3
STT-tensorflow/tensorflow/compiler/jit/kernels
History
George Karpenkov 9ca89a201c [TF2XLA] [NFC] Refactor datastructures for resource variables to not require snapshotting for compilation 
Previously, `BuildXlaCompilerArguments` required taking a snapshot of all
resource variables in order to start compiling.
In this CL it can operate with a span of pointers to resource variables instead
(we CHECK at runtime that the lock is held).
That refactoring allows to launch the XLA compilation without creating an extra
reference to the underlying Tensor of the passed resource variables.

PiperOrigin-RevId: 317706126
Change-Id: I37a97601a08f165b23b4745e1b032bf91c21c313
2020-06-22 12:55:49 -07:00
..
BUILD [XLA:GPU] Don't defer host callbacks until BlockHostUntilDone(). 2020-02-12 10:21:01 -08:00
xla_ops.cc [TF2XLA] [NFC] Refactor datastructures for resource variables to not require snapshotting for compilation 2020-06-22 12:55:49 -07:00
xla_ops.h Prefixing TensorFlow thread annotation macros with TF_. 2020-03-05 08:42:01 -08:00