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. 17 4月, 2023 1 次提交
  3. 13 1月, 2023 1 次提交
    • Z
      [inference][trt]set output data type of trt network (#49712) · 690d7a69
      Zhang Jun 提交于
      * update trt engine to set in/out data type
      
      * update
      
      * Update engine.cc
      
      * Update engine.cc
      
      * update
      
      * set engine output type before freeze the network
      
      * update
      
      * update trt autoscan ut
      
      * update
      
      * update ut
      
      * fix equal bug, update ut
      
      * fix cast and equal ut
      
      * update cast ut using TRT < 8.4
      
      * set datatype from scope
      
      * check output var is nullptr
      
      * Update op_converter.h
      
      * update tensorrt_engine_op_test ut
      
      * update
      690d7a69
  4. 25 11月, 2022 1 次提交
  5. 01 11月, 2022 1 次提交
  6. 23 10月, 2022 1 次提交
  7. 19 10月, 2022 1 次提交
  8. 12 10月, 2022 1 次提交
  9. 13 6月, 2022 1 次提交
  10. 05 6月, 2022 1 次提交
    • S
      【code format check upgrade】 step2:yapf (#42944) · a072fca8
      Sing_chan 提交于
      * use yapf to format all python file
      
      * yapf exclude two unittests file for they rely on writing and reading file, and format will break them
      
      * disable diff_py_file because too many diff files cause command following failed
      a072fca8
  11. 22 2月, 2022 1 次提交
  12. 20 10月, 2021 1 次提交
  13. 14 9月, 2021 2 次提交