STT-tensorflow/tensorflow/compiler
Yanhua Sun b2f0928940 Add DeviceIndex xla op.
DeviceIndex op: given a list of device names, this operation returns the index of the device this op runs.  In the case of XLA, we are not executing on any device, we return the length of the list.

PiperOrigin-RevId: 317740778
Change-Id: I0679aa0adc5508b83502eee0d2044584577ed5b4
2020-06-22 15:06:38 -07:00
..
aot Fix the tests that would be otherwise broken by https://reviews.llvm.org/D80258. 2020-06-15 17:15:24 -07:00
jit Add DeviceIndex xla op. 2020-06-22 15:06:38 -07:00
mlir Add quantization support for tfl.transpose_conv 2020-06-22 13:46:24 -07:00
plugin
tensorflow/tests Move tfl-device-index-selector to TF directory. 2020-06-18 15:11:43 -07:00
tests Do not try to compile trivially dead branches in the Case tf2xla lowering 2020-06-19 14:18:26 -07:00
tf2tensorrt Rollback "[TF:TRT] Cosmetic fix." 2020-06-19 17:15:15 -07:00
tf2xla Add DeviceIndex xla op. 2020-06-22 15:06:38 -07:00
xla Make XRT CPU/GPU use MaybeOwning buffer interface, so the new copy protection CL won't break aliasing. 2020-06-22 12:32:43 -07:00
xrt Make XRT CPU/GPU use MaybeOwning buffer interface, so the new copy protection CL won't break aliasing. 2020-06-22 12:32:43 -07:00