• L
    Add float16 type for scatter op. (#38136) · 9bac4a76
    Li Min 提交于
    * Add float16 type for scatter op.
    
    * Add fp16 test for scatter op.
    
    * Add int and int64 support for scatter_grad on gpu.
    
    * Add int and int64 for check_variable_and_dtype routine.
    
    * Minors.
    
    * Code format.
    9bac4a76
test_scatter_op.py 12.1 KB