Commit 53cc63a2 authored by Dhananjay Nakrani's avatar Dhananjay Nakrani Committed by TensorFlower Gardener
Browse files

[part 1] Add support for int32 & int64 in RandomPoissonOp. 

This computes int32/int64-precision poisson samples with double precision intermediate calculations (same as it's done for `half`) respectively.

part 2 will switch over python calls to new op once forward compatibility period has passed.

PiperOrigin-RevId: 171058336
parent 70fc9bf9
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