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.
 
 
 
 
 
 
ZPaC 04ee8f4dd9 Support reference for gpu and cpu for parameter update 4 years ago
..
activation_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
activation_gpu_kernel.h return default shape when max_shape is empty 4 years ago
activation_grad_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
activation_grad_kernel.h extract common as an independent shared library 4 years ago
adagrad_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
adagrad_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
adam_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
adam_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
adam_weight_decay_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
adam_weight_decay_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
adaptive_avg_pool2d_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
adaptive_avg_pool2d_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
adaptive_avg_pool2d_grad_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
adaptive_avg_pool2d_grad_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
apply_gradient_descent_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
apply_gradient_descent_gpu_kernel.h extract common as an independent shared library 4 years ago
batch_norm_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
batch_norm_gpu_kernel.h extract common as an independent shared library 4 years ago
batch_norm_grad_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
batch_norm_grad_gpu_kernel.h [DynamicShape][GPU]add dynamic shape support some ops for new network 4 years ago
bce_with_logits_loss_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
bce_with_logits_loss_kernel.h extract common as an independent shared library 4 years ago
bias_add_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
bias_add_gpu_kernel.h extract common as an independent shared library 4 years ago
bias_add_grad_gpu_kenel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
bias_add_grad_gpu_kenel.h extract common as an independent shared library 4 years ago
binary_cross_entropy_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
binary_cross_entropy_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
binary_cross_entropy_grad_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
binary_cross_entropy_grad_kernel.h delete GPU/CPUDynamicKernel 4 years ago
clip_grad_norm_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
clip_grad_norm_gpu_kernel.h extract common as an independent shared library 4 years ago
combine_momentum_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
combine_momentum_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
conv2d_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
conv2d_gpu_kernel.h extract common as an independent shared library 4 years ago
conv2d_grad_filter_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
conv2d_grad_filter_gpu_kernel.h extract common as an independent shared library 4 years ago
conv2d_grad_input_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
conv2d_grad_input_gpu_kernel.h extract common as an independent shared library 4 years ago
conv3d_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
conv3d_gpu_kernel.h extract common as an independent shared library 4 years ago
conv3d_grad_filter_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
conv3d_grad_filter_gpu_kernel.h extract common as an independent shared library 4 years ago
conv3d_grad_input_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
conv3d_grad_input_gpu_kernel.h extract common as an independent shared library 4 years ago
conv3d_transpose_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
conv3d_transpose_gpu_kernel.h extract common as an independent shared library 4 years ago
ctcloss_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
ctcloss_gpu_kernel.h extract common as an independent shared library 4 years ago
dropout3d_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
dropout3d_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
dropout_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
dropout_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
dropout_grad_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
dropout_grad_kernel.h delete GPU/CPUDynamicKernel 4 years ago
flatten_gpu_kernel.cc [DynamicShape][GPU]add dynamic shape support some ops for new network 4 years ago
flatten_gpu_kernel.h [DynamicShape][GPU]add dynamic shape support some ops for new network 4 years ago
flatten_grad_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
flatten_grad_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
ftrl_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
ftrl_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
fused_adam_weight_decay.cc Refactor ccsrc directories and CMakeLists files 4 years ago
fused_adam_weight_decay.h delete GPU/CPUDynamicKernel 4 years ago
fused_add_relu_grad_v2_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
fused_add_relu_grad_v2_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
fused_add_relu_v2_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
fused_add_relu_v2_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
fused_scale_momentum_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
fused_scale_momentum_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
fused_weightdecay_momentum_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
fused_weightdecay_momentum_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
fused_weightdecay_scale_momentum_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
fused_weightdecay_scale_momentum_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
gelu_grad_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
gelu_grad_kernel.h delete GPU/CPUDynamicKernel 4 years ago
gelu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
gelu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
hsigmoid_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
hsigmoid_gpu_kernel.h extract common as an independent shared library 4 years ago
hsigmoid_grad_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
hsigmoid_grad_gpu_kernel.h extract common as an independent shared library 4 years ago
hswish_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
hswish_gpu_kernel.h extract common as an independent shared library 4 years ago
hswish_grad_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
hswish_grad_gpu_kernel.h extract common as an independent shared library 4 years ago
im2col_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
im2col_gpu_kernel.h extract common as an independent shared library 4 years ago
instance_norm_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
instance_norm_gpu_kernel.h extract common as an independent shared library 4 years ago
instance_norm_grad_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
instance_norm_grad_gpu_kernel.h extract common as an independent shared library 4 years ago
kl_div_loss_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
kl_div_loss_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
kl_div_loss_grad_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
kl_div_loss_grad_kernel.h delete GPU/CPUDynamicKernel 4 years ago
l2_loss_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
l2_loss_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
l2normalize_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
l2normalize_gpu_kernel.h extract common as an independent shared library 4 years ago
l2normalize_grad_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
l2normalize_grad_gpu_kernel.h extract common as an independent shared library 4 years ago
layer_norm_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
layer_norm_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
layer_norm_grad_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
layer_norm_grad_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
layer_norm_grad_grad_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
layer_norm_grad_grad_gpu_kernel.h extract common as an independent shared library 4 years ago
local_response_norm_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
local_response_norm_gpu_kernel.h extract common as an independent shared library 4 years ago
local_response_norm_grad_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
local_response_norm_grad_gpu_kernel.h extract common as an independent shared library 4 years ago
lstm_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
lstm_gpu_kernel.h extract common as an independent shared library 4 years ago
lstm_grad_data_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
lstm_grad_data_gpu_kernel.h extract common as an independent shared library 4 years ago
lstm_grad_weight_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
lstm_grad_weight_gpu_kernel.h extract common as an independent shared library 4 years ago
maxpool_with_argmax_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
maxpool_with_argmax_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
maxpool_with_argmax_grad_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
maxpool_with_argmax_grad_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
mirror_pad_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
mirror_pad_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
mirror_pad_grad_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
mirror_pad_grad_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
momentum_gpu_kernel.cc Support reference for gpu and cpu for parameter update 4 years ago
momentum_gpu_kernel.h Support reference for gpu and cpu for parameter update 4 years ago
nll_loss_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
nll_loss_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
nll_loss_grad_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
nll_loss_grad_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
pad_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
pad_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
pooling_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
pooling_gpu_kernel.h extract common as an independent shared library 4 years ago
pooling_grad_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
pooling_grad_gpu_kernel.h extract common as an independent shared library 4 years ago
prelu_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
prelu_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
prelu_grad_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
prelu_grad_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
ps_roi_pooling_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
ps_roi_pooling_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
ps_roi_pooling_grad_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
ps_roi_pooling_grad_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
relu_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
relu_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
relu_grad_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
relu_grad_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
relu_grad_v2_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
relu_grad_v2_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
relu_v2_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
relu_v2_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
resize_bilinear_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
resize_bilinear_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
resize_bilinear_grad_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
resize_bilinear_grad_gpu_kernel.h extract common as an independent shared library 4 years ago
rmsprop_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
rmsprop_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
roi_align_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
roi_align_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
roi_align_grad_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
roi_align_grad_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
sgd_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
sgd_gpu_kernel.h extract common as an independent shared library 4 years ago
sigmoid_cross_entropy_with_logits_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
sigmoid_cross_entropy_with_logits_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
sigmoid_cross_entropy_with_logits_grad_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
sigmoid_cross_entropy_with_logits_grad_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
smooth_l1_loss_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
smooth_l1_loss_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
smooth_l1_loss_grad_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
smooth_l1_loss_grad_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
softmax_cross_entropy_with_logits_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
softmax_cross_entropy_with_logits_gpu_kernel.h extract common as an independent shared library 4 years ago
softmax_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
softmax_gpu_kernel.h extract common as an independent shared library 4 years ago
softmax_grad_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
softmax_grad_gpu_kernel.h extract common as an independent shared library 4 years ago
softplus_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
softplus_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
softplus_grad_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
softplus_grad_gpu_kernel.h delete GPU/CPUDynamicKernel 4 years ago
sparse_apply_proximal_adagrad_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
sparse_apply_proximal_adagrad_kernel.h delete GPU/CPUDynamicKernel 4 years ago
sparse_ftrl_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
sparse_ftrl_gpu_kernel.h extract common as an independent shared library 4 years ago
sparse_softmax_cross_entropy_with_logits_gpu_kernel.cc Refactor ccsrc directories and CMakeLists files 4 years ago
sparse_softmax_cross_entropy_with_logits_gpu_kernel.h extract common as an independent shared library 4 years ago