Add static_sample flag to Mixture, permitting calls to `sample` to not rely on...
Add static_sample flag to Mixture, permitting calls to `sample` to not rely on dynamic tensor indexing. This allows for some static graph compilation optimizations, but at the expense of sampling all underlying distributions in the mixture. PiperOrigin-RevId: 183869189
Loading
Please sign in to comment