diff options
Diffstat (limited to 'tensorflow/compiler/xla/client/lib/testing.h')
-rw-r--r-- | tensorflow/compiler/xla/client/lib/testing.h | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/client/lib/testing.h b/tensorflow/compiler/xla/client/lib/testing.h index 7e640d1307..b5c4393dcc 100644 --- a/tensorflow/compiler/xla/client/lib/testing.h +++ b/tensorflow/compiler/xla/client/lib/testing.h @@ -26,6 +26,10 @@ limitations under the License. namespace xla { +// Generates fake data in a literal of the given shape, or returns an error +// status if the element type is currently unhandled for fake data generation. +StatusOr<std::unique_ptr<Literal>> MakeFakeLiteral(const Shape& shape); + // Generates fake data of the given shape on the device or dies. The fake data // is created by performing a computation on the device rather than transferring // data from the host to the device. |