Commit 9e481837 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by Rasmus Larsen
Browse files

Disable x * x -> square(x) Grapler rewrite for complex types unless the op is... 

Disable x * x -> square(x) Grapler rewrite for complex types unless the op is on CPU. Square is not registered for complex types on GPU, and doing so produces a crash in with CUDA_ILLEGAL_INSTRUCTION when running it on open source ubuntu.

PiperOrigin-RevId: 192788160
parent e9e5356b
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