From 36647440d2e62cb494e4e6f6d5d9144ceb0b29c7 Mon Sep 17 00:00:00 2001 From: Mark Heffernan Date: Thu, 21 Sep 2017 15:05:33 -0700 Subject: 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 --- tensorflow/compiler/xla/service/local_service.h | 8 -------- 1 file changed, 8 deletions(-) (limited to 'tensorflow/compiler/xla/service/local_service.h') 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> 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 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. -- cgit v1.2.3