diff options
author | 2017-09-21 15:05:33 -0700 | |
---|---|---|
committer | 2017-09-21 15:08:28 -0700 | |
commit | 36647440d2e62cb494e4e6f6d5d9144ceb0b29c7 (patch) | |
tree | d97c715ef6c79b205442f254679b1ffa03be94e4 /tensorflow/compiler/xla/service/local_service.h | |
parent | 57498a86c11dfc98dda84dc7318a3c84c85c6791 (diff) |
Add methods to convert between Literals and ShapedBuffers to LocalClient. These "conversion" methods copy the data to/from the device into/from literals.
Also fix various issues I noticed along the way:
* Move LocalClient tests into open source.
* Add proper == operator to Literals
* Add << overload for streaming Literals to output.
* Add Literal::GetSubliteral methods.
* Remove unused AllocatBufferOnDevice methods from LocalClient and LocalService.
PiperOrigin-RevId: 169606342
Diffstat (limited to 'tensorflow/compiler/xla/service/local_service.h')
-rw-r--r-- | tensorflow/compiler/xla/service/local_service.h | 8 |
1 files changed, 0 insertions, 8 deletions
diff --git a/tensorflow/compiler/xla/service/local_service.h b/tensorflow/compiler/xla/service/local_service.h index c90943f3c0..f2bfb960f4 100644 --- a/tensorflow/compiler/xla/service/local_service.h +++ b/tensorflow/compiler/xla/service/local_service.h @@ -39,14 +39,6 @@ class LocalService : public Service { static StatusOr<std::unique_ptr<LocalService>> NewService( const ServiceOptions& options); - // Return a handle to a buffer large enough to hold shape, allocated - // on device_ordinal. If allocate_space_for_deep_copy, the buffer is - // large enough to hold all sub-buffers of a tuple shape, otherwise - // it is only as large as the top-level tuple pointer array. - StatusOr<GlobalDataHandle> AllocateBufferOnDevice( - const Shape& shape, int device_ordinal, - bool allocate_space_for_deep_copy); - // Builds an Executable with the given argument layouts and options. If // result_layout is non-null, then the executable is compiled to produce a // result of the given layout. |