• 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
range_op.cc 2.6 KB