Commit 9bac04a4 authored by Brian Patton's avatar Brian Patton Committed by TensorFlower Gardener
Browse files

Modifies kConditional to support both predicated and indexed conditionals (i.e. switch statements). 

Updates lowerings for both CPU and GPU.

Adds a new tf2xla kernel for the CFv2 functional Case op which lowers to an indexed kConditional.

PiperOrigin-RevId: 235831207
parent 39b741fd
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