1. 28 Aug, 2023 1 commit
  2. 04 Aug, 2023 1 commit
  3. 31 Jul, 2023 1 commit
  4. 27 Jul, 2023 1 commit
  5. 26 Jul, 2023 1 commit
  6. 05 Jul, 2023 1 commit
  7. 29 Jun, 2023 1 commit
  8. 22 Jun, 2022 1 commit
  9. 05 Jan, 2022 1 commit
  10. 01 Oct, 2021 1 commit
    • turneram's avatar
      Add multinomial op (#954) · 0b7672d7
      turneram authored
      
      
      Add multinomial op to onnx parser with ref and GPU implementations.
      
      The onnx parser inserts a literal of shape {batch_size, sample_size} with random values in the range [0, 1) and inserts existing ops to compute the cumulative density function. The multinomial operator multiplies the random values by the sum of the CDF and returns the index of the first element of the CDF that is greater than the result, representing samples randomly drawn from [0, class_size) that follow the log-probability distribution.
      
      Resolves #821
      Co-authored-by: default avatarShucai Xiao <shucai@gmail.com>
      0b7672d7