[XLA] Make ShapeTree use ShapeIndexViews
Avoids creating temporary std::vectors on the consumer side. Also push ShapeIndexViews through the GPU backend a bit. PiperOrigin-RevId: 201529722
Loading
Please sign in to comment
Avoids creating temporary std::vectors on the consumer side. Also push ShapeIndexViews through the GPU backend a bit. PiperOrigin-RevId: 201529722