1. 26 6月, 2023 1 次提交
    • Z
      [inference][trt] optimize set_value and top_k op (#54372) · e25e86f4
      Zhang Jun 提交于
      * set_value update
      
      * support ValueTensor's rank != Input'rank & update topk
      
      * update range to avoid coredump
      
      * fix addShape error
      
      * Dims definition differ between 7.2 and 8.0+
      
      * Update test_trt_convert_top_k_v2.py
      
      * update top_k
      
      * Update test_trt_convert_top_k_v2.py
      e25e86f4
  2. 05 12月, 2022 1 次提交