Commit b49aef3c authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Add methods RemoveLastDims and RemoveDimRange to TensorShape. Apart from the... 

Add methods RemoveLastDims and RemoveDimRange to TensorShape. Apart from the convenience, some kernels call RemoveDim multiple times, or even in a loop, which ends up being an O(dims()*k) operation instead of O(dims()).

PiperOrigin-RevId: 169473109
parent 75e7bada
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