Commit f6ee54c9 authored by Skye Wanderman-Milne's avatar Skye Wanderman-Milne Committed by TensorFlower Gardener
Browse files

cond_v2: use optional tensors instead of FakeParams. 

The purpose of this change is to not waste memory allocating large
FakeParams, which is especially important on GPU.

This also adds a few other fixes needed to get optional variants
working with cond_v2, including on GPU.

PiperOrigin-RevId: 223260005
parent 22ff3ec6
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