Commit 40ff081c authored by Bixia Zheng's avatar Bixia Zheng Committed by TensorFlower Gardener
Browse files

[XLA] Teach algebraic simplifier to convert a copy instruction to a bitcast 

instruction.

Extend ReshapeIsBitcast to handle copy instructions. This allows algebraic
simplifier, for example, to replace a copy from shape f16[1,1,128,128]{3,2,1,0}
to shape f16[1,1,128,128]{1,0,3,2} to bitcast for the CPU and GPU backends.

Add a test case.

PiperOrigin-RevId: 201699400
parent b653d1ac
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