• Y
    Optimize the elementwise op using eigen (#15494) · dcda2023
    Yiqun Liu 提交于
    * Optimize the elementwise op with CUDA kernels.
    test=develop
    
    * Support setting of attr in op config file.
    test=develop
    
    * Add the support the setting dtype and initializer in config.
    test=develop
    
    * Save workspace.
    
    * Add initializer "zeros".
    test=develop
    
    * Fix compiling error.
    
    * Support the use of existed file to initailize tensor in op_tester.
    
    * Use eigen to optimize the elementwise_add/mul for the case that x and y have the same dims.
    test=develop
    dcda2023
elementwise_add_op.h 6.3 KB