• Y
    A a pass to enable the use of cudnn (#19346) · c5548178
    Yiqun Liu 提交于
    * Add a interface to enable cudnn for inference.
    
    * Add cudnn_placement_pass.
    test=develop
    
    * Set the default value of cudnn_enabled_op_types to null.
    test=develop
    
    * Write the common basic class, placement_pass_base, to refine the codes.
    test=develop
    
    * Call EnableCUDNN in unittest.
    test=develop
    
    * Refine cudnn_placement_pass tester.
    
    * Enable the testing of cudnn_placement_pass in inference's unittest.
    test=develop
    
    * Add the check of op kernels.
    test=develop
    c5548178
trt_resnet50_test.cc 996 字节