Commit 36647440 authored by Mark Heffernan's avatar Mark Heffernan Committed by TensorFlower Gardener
Browse files

Add methods to convert between Literals and ShapedBuffers to LocalClient.... 

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
parent 57498a86
Expand all Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment