You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
zengzitao 3ef0e9f053 substitute dropout by cudnnuniformreal and dropout 5 years ago
..
__init__.py substitute dropout by cudnnuniformreal and dropout 5 years ago
bias_add.py expand bias_add and bias_add_grad op 5 years ago
bias_add_grad.py expand bias_add and bias_add_grad op 5 years ago
dropout_grad.py expand maximum_grad minimum_grad dropout_grad op 5 years ago
fused_adam.py support expand fused_adam and fused_adam_weight_decay op 5 years ago
fused_adam_weight_decay.py support expand fused_adam and fused_adam_weight_decay op 5 years ago
gelu.py declare fp32 and than cast to fp16 in expander 5 years ago
gelu_grad.py declare fp32 and than cast to fp16 in expander 5 years ago
gkdropout.py substitute dropout by cudnnuniformreal and dropout 5 years ago
layernorm.py expand logsoftmax and logsoftmax_grad, delete softmax's cast and fix layernorm op 5 years ago
layernorm_grad.py expand layernorm_grad op 5 years ago
logsoftmax.py expand logsoftmax and logsoftmax_grad, delete softmax's cast and fix layernorm op 5 years ago
logsoftmax_grad.py expand logsoftmax and logsoftmax_grad, delete softmax's cast and fix layernorm op 5 years ago
maximum_grad.py expand maximum_grad minimum_grad dropout_grad op 5 years ago
minimum_grad.py expand maximum_grad minimum_grad dropout_grad op 5 years ago
reduce_mean.py expand tanh_grad and reduce_mean, fix bug and add test_case in ci 5 years ago
softmax.py expand logsoftmax and logsoftmax_grad, delete softmax's cast and fix layernorm op 5 years ago
square.py GraphKernel supports GPU 5 years ago
tanh_grad.py expand tanh_grad and reduce_mean, fix bug and add test_case in ci 5 years ago