Commit 70d8b16a authored by Benjamin Kramer's avatar Benjamin Kramer Committed by TensorFlower Gardener
Browse files

[XLA:GPU] Pick the right shape for emitting memsets for elements with size 2 

Otherwise this could would be wrong for multi-output fusions of fp16 or bf16.
We currently never use those for reduce-fusions on GPU.

PiperOrigin-RevId: 201508558
parent dfbdc142
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