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.
 
 
 
 
 
 
lizhenyu ea0cd5ccdd add Sigmoid and SigmoidGrad operation of GPU 5 years ago
..
activation_gpu_kernel.cc add Sigmoid and SigmoidGrad operation of GPU 5 years ago
activation_gpu_kernel.h replace tanh implement with cudnn 5 years ago
activation_grad_kernel.cc add Sigmoid and SigmoidGrad operation of GPU 5 years ago
activation_grad_kernel.h replace tanh implement with cudnn 5 years ago
adam_gpu_kernel.cc add adam optimizer 5 years ago
adam_gpu_kernel.h add adam optimizer 5 years ago
bias_add_grad_gpu_kenel.cc initial version 6 years ago
bias_add_grad_gpu_kenel.h gpu refactor cuda datatype 5 years ago
conv2d_gpu_kernel.cc initial version 6 years ago
conv2d_gpu_kernel.h gpu refactor cuda datatype 5 years ago
conv2d_grad_filter_gpu_kernel.cc initial version 6 years ago
conv2d_grad_filter_gpu_kernel.h gpu refactor cuda datatype 5 years ago
conv2d_grad_input_gpu_kernel.cc initial version 6 years ago
conv2d_grad_input_gpu_kernel.h gpu refactor cuda datatype 5 years ago
dropout_gpu_kernel.cc gpu codex warning fix 5 years ago
dropout_gpu_kernel.h Gpu Dropout kernel fix 5 years ago
dropout_grad_kernel.cc gpu codex warning fix 5 years ago
dropout_grad_kernel.h Gpu Dropout kernel fix 5 years ago
flatten_gpu_kernel.cc initial version 6 years ago
flatten_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
flatten_grad_gpu_kernel.cc initial version 6 years ago
flatten_grad_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
ftrl_gpu_kernel.cc add ftrl optimizer 5 years ago
ftrl_gpu_kernel.h add ftrl optimizer 5 years ago
fused_batch_norm_gpu_kernel.cc modify ResizeNearestNeighborV2D 5 years ago
fused_batch_norm_gpu_kernel.h gpu refactor cuda datatype 5 years ago
fused_batchnorm_grad_gpu_kernel.cc initial version 6 years ago
fused_batchnorm_grad_gpu_kernel.h gpu refactor cuda datatype 5 years ago
gelu_grad_kernel.cc gpu support Gelu & GeluGrad kernels 5 years ago
gelu_grad_kernel.h Change uintptr_t to void ptr 5 years ago
gelu_kernel.cc gpu support Gelu & GeluGrad kernels 5 years ago
gelu_kernel.h Change uintptr_t to void ptr 5 years ago
layer_norm_gpu_kernel.cc Gpu support LayerNorm kernel 5 years ago
layer_norm_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
layer_norm_grad_gpu_kernel.cc Gpu support LayerNorm kernel 5 years ago
layer_norm_grad_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
lstm_gpu_kernel.cc initial version 6 years ago
lstm_gpu_kernel.h gpu refactor cuda datatype 5 years ago
lstm_grad_data_gpu_kernel.cc initial version 6 years ago
lstm_grad_data_gpu_kernel.h gpu refactor cuda datatype 5 years ago
lstm_grad_weight_gpu_kernel.cc initial version 6 years ago
lstm_grad_weight_gpu_kernel.h gpu refactor cuda datatype 5 years ago
momentum_gpu_kernel.cc initial version 6 years ago
momentum_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
pooling_gpu_kernel.cc initial version 6 years ago
pooling_gpu_kernel.h gpu refactor cuda datatype 5 years ago
pooling_grad_gpu_kernel.cc gpu codex fix 5 years ago
pooling_grad_gpu_kernel.h gpu codex fix 5 years ago
rmsprop_gpu_kernel.cc fixed validator for CumProd, ReduceProd, ApplyRMSProp 5 years ago
rmsprop_gpu_kernel.h fixed validator for CumProd, ReduceProd, ApplyRMSProp 5 years ago
sigmoid_cross_entropy_with_logits_gpu_kernel.cc add SigmoidCrossEntropyWithLogits op 5 years ago
sigmoid_cross_entropy_with_logits_gpu_kernel.h add SigmoidCrossEntropyWithLogits op 5 years ago
sigmoid_cross_entropy_with_logits_grad_gpu_kernel.cc add SigmoidCrossEntropyWithLogitsGrad op 5 years ago
sigmoid_cross_entropy_with_logits_grad_gpu_kernel.h add SigmoidCrossEntropyWithLogitsGrad op 5 years ago
softmax_cross_entropy_with_logits_gpu_kernel.cc initial version 6 years ago
softmax_cross_entropy_with_logits_gpu_kernel.h gpu refactor cuda datatype 5 years ago
softmax_gpu_kernel.cc gpu support logsoftmax & logsoftmaxgrad kernel 5 years ago
softmax_gpu_kernel.h gpu refactor cuda datatype 5 years ago
softmax_grad_gpu_kernel.cc gpu support logsoftmax & logsoftmaxgrad kernel 5 years ago
softmax_grad_gpu_kernel.h gpu refactor cuda datatype 5 years ago
sparse_softmax_cross_entropy_with_logits_gpu_kernel.cc initial version 6 years ago
sparse_softmax_cross_entropy_with_logits_gpu_kernel.h gpu refactor cuda datatype 5 years ago