• Y
    Support generating code for grad_op (#21066) · 9091f8cd
    Yiqun Liu 提交于
    * Add the definition of operation in fusion_group.
    
    * Use operations in OperationMap to detect fusion_group of elementwise pattern.
    
    * Add namespace fusion_group in code_generator.
    
    * Use operations recorded in OperationMap to generate code.
    
    * Remove implementation codes to .cc file.
    
    * Refine Operation and CodeGenerator to make it easier to generate code for grad_op.
    Refine the unittest for better reuse.
    
    * Avoid recording the template's keyword in a array.
    
    * Support the generating of code for grad_op and add unittest.
    test=develop
    
    * Remove replaced_element_in_order and use use number instead.
    test=develop
    9091f8cd
operation.cc 3.6 KB