Also:

  
      github.com/tensorflow/tensorflow
  
  
    
  
    
  

  
    
      Fix dtype mismatches in rejection_resample. 10
    

    
      
        committed 
        
          
        
        May 11, 2021
      
      

      
        
          
          MarkDaoust
        
        
      

      
        
          +34
          -3
        
      
    

  



  
    Cast `initial_dist` and `target_dist` to `float32`.
Normally TensorFlow doesn't cast things for you, but I think in this case it's reasonable because:
-...