Introduced tool to run an HLO module in replicated fashion, by infeeding random data...
authorA. Unique TensorFlower <gardener@tensorflow.org>
Thu, 12 Apr 2018 17:14:02 +0000 (10:14 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Thu, 12 Apr 2018 17:16:34 +0000 (10:16 -0700)
commitffbf77de81d0b7b4b169c92d0d9fbbdef5b8842a
tree8d68eedf28bdcac55516b6a3a176d56f6cef0fa2
parent8a247976484173059aedc17bfd8d770b8d1a70e1
Introduced tool to run an HLO module in replicated fashion, by infeeding random data and outfeeding the data generated at each step.
The arguments of the computation can be either read from the session module, or randomly generated.
The tool uses the raw transfer manager API to infeed and outfeed the data.

PiperOrigin-RevId: 192628605
tensorflow/compiler/xla/service/BUILD
tensorflow/compiler/xla/service/hlo_runner.cc
tensorflow/compiler/xla/service/hlo_runner.h
tensorflow/compiler/xla/shape_util.h
tensorflow/compiler/xla/tests/test_utils.cc
tensorflow/compiler/xla/tests/test_utils.h