Commit 47fcca75 authored by Peter Hawkins's avatar Peter Hawkins Committed by TensorFlower Gardener
Browse files

[TF:XLA] Implement MatrixSetDiag and MatrixBandPart. 

Add support for int32 indices to the MatrixBandPart operator.

PiperOrigin-RevId: 184133343
parent baf490ba
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