• 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
engine.h 30.3 KB