| .. |
|
detail
|
…
|
|
|
images
|
…
|
|
|
math
|
Merge pull request #5130 from chengduoZH/fix_sequence_conv_op
|
7 years ago |
|
nccl
|
…
|
|
|
rnn
|
…
|
|
|
.clang-format
|
…
|
|
|
CMakeLists.txt
|
Merge pull request #4838 from dzhwinter/feature/multigpu
|
7 years ago |
|
accuracy_op.cc
|
Refine evaluator op types (#5208)
|
7 years ago |
|
accuracy_op.cu
|
Refine evaluator op types (#5208)
|
7 years ago |
|
accuracy_op.h
|
Refine evaluator op types (#5208)
|
7 years ago |
|
activation_op.cc
|
…
|
|
|
activation_op.cu
|
…
|
|
|
activation_op.h
|
refine square_error_cost layer (#5216)
|
7 years ago |
|
adadelta_op.cc
|
…
|
|
|
adadelta_op.cu
|
…
|
|
|
adadelta_op.h
|
…
|
|
|
adagrad_op.cc
|
…
|
|
|
adagrad_op.cu
|
…
|
|
|
adagrad_op.h
|
…
|
|
|
adam_op.cc
|
…
|
|
|
adam_op.cu
|
…
|
|
|
adam_op.h
|
…
|
|
|
adamax_op.cc
|
…
|
|
|
adamax_op.cu
|
…
|
|
|
adamax_op.h
|
…
|
|
|
auc_op.cc
|
Refine evaluator op types (#5208)
|
7 years ago |
|
auc_op.h
|
Refine evaluator op types (#5208)
|
7 years ago |
|
batch_norm_op.cc
|
03 image classification (#5192)
|
7 years ago |
|
batch_norm_op.cu
|
batch norm fully tortured and passed
|
7 years ago |
|
batch_norm_op.h
|
…
|
|
|
batch_norm_op.md
|
…
|
|
|
cast_op.cc
|
Cast Operator (#5149)
|
7 years ago |
|
cast_op.cu
|
Cast Operator (#5149)
|
7 years ago |
|
cast_op.h
|
Cast Operator (#5149)
|
7 years ago |
|
clip_op.cc
|
…
|
|
|
clip_op.cu
|
…
|
|
|
clip_op.h
|
…
|
|
|
concat_op.cc
|
…
|
|
|
concat_op.cu
|
…
|
|
|
concat_op.h
|
…
|
|
|
cond_op.cc
|
…
|
|
|
cond_op.h
|
…
|
|
|
conv2d_op.cc
|
…
|
|
|
conv2d_op.cu
|
…
|
|
|
conv2d_op.h
|
…
|
|
|
conv2dtranspose_op.cc
|
…
|
|
|
conv2dtranspose_op.cu
|
…
|
|
|
conv2dtranspose_op.h
|
…
|
|
|
conv_cudnn_op.cc
|
…
|
|
|
conv_cudnn_op.cu
|
follow comments
|
7 years ago |
|
conv_shift_op.cc
|
…
|
|
|
conv_shift_op.cu
|
…
|
|
|
conv_shift_op.h
|
…
|
|
|
cos_sim_op.cc
|
…
|
|
|
cos_sim_op.cu
|
…
|
|
|
cos_sim_op.h
|
…
|
|
|
crop_op.cc
|
…
|
|
|
crop_op.cu
|
…
|
|
|
crop_op.h
|
…
|
|
|
cross_entropy_op.cc
|
…
|
|
|
cross_entropy_op.cu
|
support sparse output for lookup table grad op (#5145)
|
7 years ago |
|
cross_entropy_op.h
|
support sparse output for lookup table grad op (#5145)
|
7 years ago |
|
decayed_adagrad_op.cc
|
…
|
|
|
decayed_adagrad_op.cu
|
…
|
|
|
decayed_adagrad_op.h
|
…
|
|
|
dropout_op.cc
|
…
|
|
|
dropout_op.cu
|
…
|
|
|
dropout_op.h
|
…
|
|
|
dynamic_recurrent_op.cc
|
…
|
|
|
dynamic_recurrent_op.h
|
…
|
|
|
dynamic_recurrent_op_test.cc
|
…
|
|
|
elementwise_add_op.cc
|
…
|
|
|
elementwise_add_op.cu
|
…
|
|
|
elementwise_add_op.h
|
…
|
|
|
elementwise_div_op.cc
|
…
|
|
|
elementwise_div_op.cu
|
…
|
|
|
elementwise_div_op.h
|
…
|
|
|
elementwise_mul_op.cc
|
…
|
|
|
elementwise_mul_op.cu
|
…
|
|
|
elementwise_mul_op.h
|
…
|
|
|
elementwise_op.h
|
…
|
|
|
elementwise_op_function.h
|
…
|
|
|
elementwise_sub_op.cc
|
…
|
|
|
elementwise_sub_op.cu
|
…
|
|
|
elementwise_sub_op.h
|
…
|
|
|
feed_op.cc
|
support sparse output for lookup table grad op (#5145)
|
7 years ago |
|
fetch_op.cc
|
…
|
|
|
fill_constant_batch_size_like_op.cc
|
…
|
|
|
fill_constant_batch_size_like_op.cu
|
…
|
|
|
fill_constant_batch_size_like_op.h
|
…
|
|
|
fill_constant_op.cc
|
…
|
|
|
fill_constant_op.cu
|
…
|
|
|
fill_constant_op.h
|
…
|
|
|
fill_zeros_like_op.cc
|
…
|
|
|
fill_zeros_like_op.cu
|
…
|
|
|
fill_zeros_like_op.h
|
…
|
|
|
gather.cu.h
|
…
|
|
|
gather.h
|
…
|
|
|
gather_op.cc
|
…
|
|
|
gather_op.cu
|
…
|
|
|
gather_op.h
|
…
|
|
|
gather_test.cc
|
…
|
|
|
gaussian_random_op.cc
|
Adding Normal distribution initializer and unit tests for python initializers (#5256)
|
7 years ago |
|
gaussian_random_op.cu
|
…
|
|
|
gru_unit_op.cc
|
…
|
|
|
gru_unit_op.cu
|
…
|
|
|
gru_unit_op.h
|
…
|
|
|
huber_loss_op.cc
|
…
|
|
|
huber_loss_op.cu
|
…
|
|
|
huber_loss_op.h
|
Using static_cast to make more robust.
|
7 years ago |
|
increment_op.cc
|
…
|
|
|
increment_op.cu
|
…
|
|
|
increment_op.h
|
…
|
|
|
l1_norm_op.cc
|
…
|
|
|
l1_norm_op.cu
|
…
|
|
|
l1_norm_op.h
|
…
|
|
|
load_op.cc
|
…
|
|
|
lookup_table_op.cc
|
support sparse output for lookup table grad op (#5145)
|
7 years ago |
|
lookup_table_op.cu
|
support sparse output for lookup table grad op (#5145)
|
7 years ago |
|
lookup_table_op.h
|
support sparse output for lookup table grad op (#5145)
|
7 years ago |
|
lrn_op.cc
|
…
|
|
|
lrn_op.cu
|
…
|
|
|
lrn_op.h
|
…
|
|
|
lstm_op.cc
|
…
|
|
|
lstm_op.cu
|
…
|
|
|
lstm_op.h
|
…
|
|
|
lstm_unit_op.cc
|
…
|
|
|
lstm_unit_op.cu
|
…
|
|
|
lstm_unit_op.h
|
…
|
|
|
margin_rank_loss_op.cc
|
…
|
|
|
margin_rank_loss_op.cu
|
…
|
|
|
margin_rank_loss_op.h
|
…
|
|
|
matmul_op.cc
|
…
|
|
|
matmul_op.cu
|
…
|
|
|
matmul_op.h
|
…
|
|
|
mean_op.cc
|
…
|
|
|
mean_op.cu
|
…
|
|
|
mean_op.h
|
…
|
|
|
minus_op.cc
|
…
|
|
|
minus_op.cu
|
…
|
|
|
minus_op.h
|
…
|
|
|
modified_huber_loss_op.cc
|
…
|
|
|
modified_huber_loss_op.cu
|
…
|
|
|
modified_huber_loss_op.h
|
…
|
|
|
momentum_op.cc
|
…
|
|
|
momentum_op.cu
|
…
|
|
|
momentum_op.h
|
…
|
|
|
mul_op.cc
|
…
|
|
|
mul_op.cu
|
…
|
|
|
mul_op.h
|
…
|
|
|
multiplex_op.cc
|
…
|
|
|
multiplex_op.cu
|
…
|
|
|
multiplex_op.h
|
…
|
|
|
name_convention.md
|
…
|
|
|
nccl_op.cc
|
"polish code based on comment"
|
7 years ago |
|
nccl_op.cu
|
"polish code based on comment"
|
7 years ago |
|
nccl_op_test.cu
|
"polish code based on comment"
|
7 years ago |
|
net_op.cc
|
…
|
|
|
net_op.h
|
…
|
|
|
net_op_design.md
|
…
|
|
|
net_op_test.cc
|
…
|
|
|
pad_op.cc
|
…
|
|
|
pad_op.cu
|
…
|
|
|
pad_op.h
|
…
|
|
|
pool_cudnn_op.cc
|
…
|
|
|
pool_cudnn_op.cu
|
fix bug
|
7 years ago |
|
pool_cudnn_op.h
|
follow comments
|
7 years ago |
|
pool_op.cc
|
fix bug
|
7 years ago |
|
pool_op.cu
|
…
|
|
|
pool_op.h
|
fix bug
|
7 years ago |
|
pool_with_index_op.cc
|
fix bug
|
7 years ago |
|
pool_with_index_op.cu
|
…
|
|
|
pool_with_index_op.h
|
fix bug
|
7 years ago |
|
prelu_op.cc
|
…
|
|
|
prelu_op.cu
|
…
|
|
|
prelu_op.h
|
…
|
|
|
proximal_adagrad_op.cc
|
…
|
|
|
proximal_adagrad_op.cu
|
…
|
|
|
proximal_adagrad_op.h
|
…
|
|
|
proximal_gd_op.cc
|
…
|
|
|
proximal_gd_op.cu
|
…
|
|
|
proximal_gd_op.h
|
…
|
|
|
rank_loss_op.cc
|
…
|
|
|
rank_loss_op.cu
|
…
|
|
|
rank_loss_op.h
|
…
|
|
|
recurrent_op.cc
|
…
|
|
|
recurrent_op.h
|
…
|
|
|
reduce_op.cc
|
…
|
|
|
reduce_op.cu
|
…
|
|
|
reduce_op.h
|
…
|
|
|
reshape_op.cc
|
03 image classification (#5192)
|
7 years ago |
|
reshape_op.cu
|
…
|
|
|
reshape_op.h
|
03 image classification (#5192)
|
7 years ago |
|
rmsprop_op.cc
|
…
|
|
|
rmsprop_op.cu
|
…
|
|
|
rmsprop_op.h
|
…
|
|
|
rnn_design.md
|
…
|
|
|
save_load_op_test.cc
|
…
|
|
|
save_op.cc
|
…
|
|
|
scale_op.cc
|
…
|
|
|
scale_op.cu
|
…
|
|
|
scale_op.h
|
…
|
|
|
scatter.cu.h
|
…
|
|
|
scatter.h
|
…
|
|
|
scatter_op.cc
|
…
|
|
|
scatter_op.cu
|
…
|
|
|
scatter_op.h
|
…
|
|
|
scatter_test.cc
|
…
|
|
|
seq_expand_op.cc
|
Fix comments
|
7 years ago |
|
seq_expand_op.cu
|
…
|
|
|
seq_expand_op.h
|
Fix MacOS Compile (#5217)
|
7 years ago |
|
sequence_concat_op.cc
|
…
|
|
|
sequence_concat_op.cu
|
…
|
|
|
sequence_concat_op.h
|
…
|
|
|
sequence_conv_op.cc
|
fix code format and doc
|
7 years ago |
|
sequence_conv_op.cu
|
…
|
|
|
sequence_conv_op.h
|
fix code format and doc
|
7 years ago |
|
sequence_pool_op.cc
|
…
|
|
|
sequence_pool_op.cu
|
…
|
|
|
sequence_pool_op.h
|
Fix clang compile (#5171)
|
7 years ago |
|
sequence_softmax_op.cc
|
…
|
|
|
sequence_softmax_op.cu
|
…
|
|
|
sequence_softmax_op.h
|
…
|
|
|
sgd_op.cc
|
support sparse output for lookup table grad op (#5145)
|
7 years ago |
|
sgd_op.cu
|
support sparse output for lookup table grad op (#5145)
|
7 years ago |
|
sgd_op.h
|
…
|
|
|
sigmoid_cross_entropy_with_logits_op.cc
|
…
|
|
|
sigmoid_cross_entropy_with_logits_op.cu
|
…
|
|
|
sigmoid_cross_entropy_with_logits_op.h
|
…
|
|
|
sign_op.cc
|
Adding the Sign Op for L1 Weight Decay Regularization (#5138)
|
7 years ago |
|
sign_op.cu
|
Adding the Sign Op for L1 Weight Decay Regularization (#5138)
|
7 years ago |
|
sign_op.h
|
Adding the Sign Op for L1 Weight Decay Regularization (#5138)
|
7 years ago |
|
smooth_l1_loss_op.cc
|
…
|
|
|
smooth_l1_loss_op.cu
|
…
|
|
|
smooth_l1_loss_op.h
|
…
|
|
|
softmax_op.cc
|
…
|
|
|
softmax_op.cu
|
…
|
|
|
softmax_op.h
|
…
|
|
|
softmax_with_cross_entropy_op.cc
|
…
|
|
|
softmax_with_cross_entropy_op.cu
|
…
|
|
|
softmax_with_cross_entropy_op.h
|
…
|
|
|
split_op.cc
|
…
|
|
|
split_op.cu
|
…
|
|
|
split_op.h
|
…
|
|
|
squared_l2_distance_op.cc
|
…
|
|
|
squared_l2_distance_op.cu
|
…
|
|
|
squared_l2_distance_op.h
|
…
|
|
|
squared_l2_norm_op.cc
|
…
|
|
|
squared_l2_norm_op.cu
|
…
|
|
|
squared_l2_norm_op.h
|
…
|
|
|
strided_memcpy.h
|
…
|
|
|
strided_memcpy_test.cc
|
…
|
|
|
sum_op.cc
|
…
|
|
|
sum_op.cu
|
…
|
|
|
sum_op.h
|
support sparse output for lookup table grad op (#5145)
|
7 years ago |
|
top_k_op.cc
|
Add accuracy layer (#4958)
|
7 years ago |
|
top_k_op.cu
|
Fix top k op GPU code (#5221)
|
7 years ago |
|
top_k_op.h
|
Fix a type error top_k_op (#5201)
|
7 years ago |
|
transpose_op.cc
|
…
|
|
|
transpose_op.cu
|
…
|
|
|
transpose_op.h
|
…
|
|
|
uniform_random_op.cc
|
support sparse output for lookup table grad op (#5145)
|
7 years ago |
|
uniform_random_op.cu
|
support sparse output for lookup table grad op (#5145)
|
7 years ago |
Abhinav Arora