Change output type of LogProbFn from Logit = log(p)/log(1-p) to LogProb
Change output type of LogProbFn from Logit = log(p)/log(1-p) to LogProb

PiperOrigin-RevId: 418780267
Change-Id: Icc1fb76ee1186ba4c11399199995be53a889b2d7

deepmind/acme,deepmind/acme