Commit 5f1379e4 authored by Peter Hawkins's avatar Peter Hawkins Committed by TensorFlower Gardener
Browse files

[XLA] Move ExtractSubShapedBuffer into xla::ScopedShapedBuffer as a new TakeSubTree() method. 

This method seems like it more naturally belongs in the XLA libraries.

Simplify and generalize the implementation and add a test.

PiperOrigin-RevId: 220343391
parent 3908ccbc
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