Commit 555bcc14 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Allow for logit_fn's to return dictionaries of Tensors in caller utility for logit_fn. 

In dnn_logit_fn builder, do not allow for list of units -- instead, return a single Tensor.  MultiHead will handle this through logit splitting (and also handles dictionary returns for other logit_fn's).

PiperOrigin-RevId: 174522912
parent 88338151
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