Commit 253bcbb7 authored by Kay Zhu's avatar Kay Zhu Committed by TensorFlower Gardener
Browse files

[XLA] Use HloEvaluator for convolution in reference_util. 

Also Speed up HloEvaluator's HandleConvolution in non-opt build, by moving calls
to HloInstruction::shape() out of the inner loop.

PiperOrigin-RevId: 163416183
parent 569a00e6
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