1. 22 10月, 2022 1 次提交
    • A
      Fix cwise dimension overflow issue again. · 05521499
      Antonio Sanchez 提交于
      If resulting dimensions overflow an int32, we were seeing an overflow and
      crash due to size mismatch during broadcast assignment.  The cause is a simple
      dimension type mismatch.
      
      Note that actual tests for this are currently impractical, since successful
      operations require more than 2^32 elements and OOM on most machines.
      
      PiperOrigin-RevId: 479336566
      05521499
  2. 21 10月, 2022 4 次提交
  3. 13 10月, 2022 3 次提交
  4. 12 10月, 2022 4 次提交
    • L
      Merge pull request #58056 from... · 9a7518ce
      learning-to-play 提交于
      Merge pull request #58056 from vinila21/cherrypick-f5381e0e-on-r2.10
      
      Fix OOB error when op input sizes do not match.
      9a7518ce
    • L
      Merge pull request #58060 from... · 422d24d3
      learning-to-play 提交于
      Merge pull request #58060 from vinila21/cherrypick-b389f5c9-on-r2.10
      
      Add true_classes input validation for candidate sampler ops.
      422d24d3
    • A
      Add true_classes input validation for candidate sampler ops. · ee8c6907
      Antonio Sanchez 提交于
      The values must be within the valid range of the sampler.  Added a
      check for this.
      
      PiperOrigin-RevId: 479441496
      ee8c6907
    • A
      Fix OOB error when op input sizes do not match. · 1bd22f0f
      Antonio Sanchez 提交于
      In cases where op input sizes are specified as in
      ```
      REGISTER_OP("DynamicStitch")
          .Input("indices: N * int32")
          .Input("data: N * T")
          .Output("merged: T")
          .Attr("N : int >= 1")
          .Attr("T : type")
          .SetShapeFn(DynamicStitchShapeFunction);
      ```
      if differing number of inputs are provided (e.g. 3 for `indices` and 4 for `data`)
      we can get a crash in the executor when parsing the inputs, even before the kernel
      called.  Here we avoid this by checking the return code for the argument id and
      exit early.
      
      PiperOrigin-RevId: 478068540
      1bd22f0f
  5. 08 10月, 2022 2 次提交
  6. 07 10月, 2022 5 次提交
  7. 05 10月, 2022 2 次提交
  8. 04 10月, 2022 1 次提交
  9. 30 9月, 2022 1 次提交
  10. 29 9月, 2022 2 次提交
  11. 28 9月, 2022 3 次提交
  12. 23 9月, 2022 1 次提交
  13. 03 9月, 2022 4 次提交
  14. 30 8月, 2022 2 次提交
  15. 26 8月, 2022 5 次提交