• X
    【PRIM】Support use operator's output metadata info in constructing static... · d8643cb6
    Xiaoxu Chen 提交于
    【PRIM】Support use operator's output metadata info  in constructing static backward composite (#50043)
    
    * [prim] support custom target_gradients
    
    * support infershape after append one gradop
    
    * [prim] add simple net test
    
    * fix test_loop segment fault bug
    
    * [prim] fix infer shape segment fault bug when output of grad_op_desc is empty
    d8643cb6
translated_layer.py 61.1 KB