.. |
anakin
|
…
|
|
benchmark
|
…
|
|
controlflow
|
…
|
|
detail
|
…
|
|
detection
|
…
|
|
distributed
|
…
|
|
distributed_ops
|
…
|
|
elementwise
|
…
|
|
fused
|
…
|
|
jit
|
…
|
|
math
|
…
|
|
metrics
|
…
|
|
mkldnn
|
…
|
|
nccl
|
…
|
|
ngraph
|
…
|
|
optimizers
|
…
|
|
reader
|
Clean build strategy (#18148)
|
6 years ago |
reduce_ops
|
…
|
|
sequence_ops
|
Fix the bug of sequence_unpad op (#18290)
|
6 years ago |
tensorrt
|
…
|
|
CMakeLists.txt
|
…
|
|
activation_cudnn.cu.cc
|
…
|
|
activation_cudnn_op.cu.cc
|
…
|
|
activation_op.cc
|
…
|
|
activation_op.cu
|
…
|
|
activation_op.h
|
…
|
|
add_position_encoding_op.cc
|
…
|
|
add_position_encoding_op.h
|
…
|
|
affine_channel_op.cc
|
…
|
|
affine_channel_op.cu
|
…
|
|
affine_grid_cudnn_op.cu.cc
|
…
|
|
affine_grid_op.cc
|
…
|
|
affine_grid_op.h
|
…
|
|
alloc_continuous_space_op.cc
|
…
|
|
arg_max_op.cc
|
…
|
|
arg_max_op.cu
|
…
|
|
arg_min_max_op_base.h
|
…
|
|
arg_min_op.cc
|
…
|
|
arg_min_op.cu
|
…
|
|
argsort_op.cc
|
…
|
|
argsort_op.cu
|
…
|
|
argsort_op.h
|
…
|
|
array_operator.h
|
…
|
|
array_to_lod_tensor_op.cc
|
…
|
|
assign_op.cc
|
…
|
|
assign_value_op.cc
|
…
|
|
assign_value_op.cu.cc
|
…
|
|
assign_value_op.h
|
…
|
|
attention_lstm_op.cc
|
…
|
|
attention_lstm_op.h
|
…
|
|
average_accumulates_op.cc
|
…
|
|
average_accumulates_op.cu
|
…
|
|
average_accumulates_op.h
|
…
|
|
batch_norm_op.cc
|
…
|
|
batch_norm_op.cu
|
…
|
|
batch_norm_op.h
|
…
|
|
batch_size_like.h
|
…
|
|
beam_search_decode_op.cc
|
…
|
|
beam_search_decode_op.h
|
…
|
|
beam_search_decode_op_test.cc
|
…
|
|
beam_search_op.cc
|
…
|
|
beam_search_op.cu.cc
|
…
|
|
beam_search_op.h
|
…
|
|
bilinear_tensor_product_op.cc
|
…
|
|
bilinear_tensor_product_op.cu
|
…
|
|
bilinear_tensor_product_op.h
|
…
|
|
bpr_loss_op.cc
|
…
|
|
bpr_loss_op.h
|
…
|
|
cast_op.cc
|
…
|
|
cast_op.cu
|
…
|
|
cast_op.h
|
…
|
|
chunk_eval_op.cc
|
…
|
|
chunk_eval_op.h
|
…
|
|
clip_by_norm_op.cc
|
…
|
|
clip_by_norm_op.cu
|
…
|
|
clip_by_norm_op.h
|
…
|
|
clip_op.cc
|
…
|
|
clip_op.cu
|
…
|
|
clip_op.h
|
…
|
|
concat_op.cc
|
…
|
|
concat_op.cu.cc
|
…
|
|
concat_op.h
|
…
|
|
conv_cudnn_helper.h
|
…
|
|
conv_cudnn_op.cu.cc
|
…
|
|
conv_cudnn_op_cache.h
|
…
|
|
conv_fusion_op.cc
|
…
|
|
conv_fusion_op.cu.cc
|
…
|
|
conv_op.cc
|
…
|
|
conv_op.cu.cc
|
…
|
|
conv_op.h
|
…
|
|
conv_shift_op.cc
|
…
|
|
conv_shift_op.cu
|
…
|
|
conv_shift_op.h
|
…
|
|
conv_transpose_cudnn_op.cu.cc
|
…
|
|
conv_transpose_op.cc
|
…
|
|
conv_transpose_op.cu.cc
|
…
|
|
conv_transpose_op.h
|
…
|
|
cos_sim_op.cc
|
…
|
|
cos_sim_op.cu
|
…
|
|
cos_sim_op.h
|
…
|
|
crf_decoding_op.cc
|
…
|
|
crf_decoding_op.h
|
…
|
|
crop_op.cc
|
…
|
|
crop_op.cu
|
…
|
|
crop_op.h
|
…
|
|
cross_entropy_op.cc
|
…
|
|
cross_entropy_op.cu
|
…
|
|
cross_entropy_op.h
|
…
|
|
ctc_align_op.cc
|
…
|
|
ctc_align_op.cu
|
…
|
|
ctc_align_op.h
|
…
|
|
cudnn_lstm_op.cc
|
…
|
|
cudnn_lstm_op.cu.cc
|
…
|
|
cudnn_rnn_cache.h
|
…
|
|
cum_op.h
|
…
|
|
cumsum_op.cc
|
…
|
|
cumsum_op.cu
|
…
|
|
cvm_op.cc
|
…
|
|
cvm_op.h
|
…
|
|
data_norm_op.cc
|
…
|
|
data_norm_op.h
|
…
|
|
deformable_conv_op.cc
|
…
|
|
deformable_conv_op.cu
|
…
|
|
deformable_psroi_pooling_op.cc
|
…
|
|
deformable_psroi_pooling_op.cu
|
…
|
|
deformable_psroi_pooling_op.h
|
…
|
|
delete_var_op.cc
|
…
|
|
dequantize_op.cc
|
…
|
|
dequantize_op.h
|
…
|
|
detection_map_op.cc
|
…
|
|
detection_map_op.h
|
…
|
|
dgc_clip_by_norm_op.cc
|
…
|
|
dgc_clip_by_norm_op.cu
|
…
|
|
dgc_clip_by_norm_op.h
|
…
|
|
dgc_op.cc
|
…
|
|
dgc_op.cu
|
…
|
|
dgc_op.h
|
…
|
|
diag_op.cc
|
…
|
|
diag_op.cu
|
…
|
|
diag_op.h
|
…
|
|
dropout_op.cc
|
…
|
|
dropout_op.cu
|
…
|
|
dropout_op.h
|
…
|
|
dropout_op_test.cc
|
…
|
|
edit_distance_op.cc
|
…
|
|
edit_distance_op.cu
|
…
|
|
edit_distance_op.h
|
…
|
|
expand_op.cc
|
…
|
|
expand_op.cu
|
…
|
|
expand_op.h
|
…
|
|
fake_dequantize_op.cc
|
…
|
|
fake_dequantize_op.cu
|
…
|
|
fake_dequantize_op.h
|
…
|
|
fake_quantize_op.cc
|
…
|
|
fake_quantize_op.cu
|
…
|
|
fake_quantize_op.h
|
…
|
|
fc_op.cc
|
…
|
|
fc_op.h
|
…
|
|
fill_constant_batch_size_like_op.cc
|
…
|
|
fill_constant_batch_size_like_op.cu.cc
|
…
|
|
fill_constant_batch_size_like_op.h
|
…
|
|
fill_constant_op.cc
|
…
|
|
fill_constant_op.cu.cc
|
…
|
|
fill_constant_op.h
|
…
|
|
fill_op.cc
|
…
|
|
fill_zeros_like_op.cc
|
…
|
|
fill_zeros_like_op.cu.cc
|
…
|
|
fill_zeros_like_op.h
|
…
|
|
flatten_op.cc
|
…
|
|
fsp_op.cc
|
…
|
|
fsp_op.cu
|
…
|
|
fsp_op.h
|
…
|
|
gather.cu.h
|
…
|
|
gather.h
|
…
|
|
gather_op.cc
|
…
|
|
gather_op.cu
|
…
|
|
gather_op.h
|
…
|
|
gather_test.cc
|
…
|
|
gaussian_random_batch_size_like_op.cc
|
…
|
|
gaussian_random_op.cc
|
…
|
|
gaussian_random_op.cu
|
…
|
|
get_tensor_from_selected_rows_op.cc
|
…
|
|
grid_sampler_cudnn_op.cu.cc
|
…
|
|
grid_sampler_op.cc
|
…
|
|
grid_sampler_op.h
|
…
|
|
group_norm_op.cc
|
…
|
|
group_norm_op.cu
|
…
|
|
group_norm_op.h
|
…
|
|
gru_op.cc
|
…
|
|
gru_op.cu.cc
|
…
|
|
gru_op.h
|
…
|
|
gru_unit_op.cc
|
…
|
|
gru_unit_op.cu
|
…
|
|
gru_unit_op.h
|
…
|
|
hash_op.cc
|
…
|
|
hash_op.h
|
…
|
|
hierarchical_sigmoid_op.cc
|
…
|
|
hierarchical_sigmoid_op.h
|
…
|
|
hinge_loss_op.cc
|
…
|
|
hinge_loss_op.cu
|
…
|
|
hinge_loss_op.h
|
…
|
|
huber_loss_op.cc
|
…
|
|
huber_loss_op.cu
|
…
|
|
huber_loss_op.h
|
…
|
|
im2sequence_op.cc
|
…
|
|
im2sequence_op.cu
|
…
|
|
im2sequence_op.h
|
…
|
|
increment_op.cc
|
…
|
|
increment_op.cu
|
…
|
|
increment_op.h
|
…
|
|
interpolate_op.cc
|
…
|
|
interpolate_op.cu
|
set src_idx > 0 for bilinear_interp_op (#18238)
|
6 years ago |
interpolate_op.h
|
set src_idx > 0 for bilinear_interp_op (#18238)
|
6 years ago |
is_empty_op.cc
|
…
|
|
is_empty_op.cu.cc
|
…
|
|
is_empty_op.h
|
…
|
|
isfinite_op.cc
|
…
|
|
isfinite_op.cu
|
…
|
|
isfinite_op.h
|
…
|
|
kldiv_loss_op.cc
|
…
|
|
kldiv_loss_op.cu
|
…
|
|
kldiv_loss_op.h
|
…
|
|
l1_norm_op.cc
|
…
|
|
l1_norm_op.cu
|
…
|
|
l1_norm_op.h
|
…
|
|
label_smooth_op.cc
|
…
|
|
label_smooth_op.cu
|
…
|
|
label_smooth_op.h
|
…
|
|
layer_norm_op.cc
|
…
|
|
layer_norm_op.cu
|
…
|
|
layer_norm_op.h
|
…
|
|
linear_chain_crf_op.cc
|
…
|
|
linear_chain_crf_op.cu
|
…
|
|
linear_chain_crf_op.h
|
…
|
|
linspace_op.cc
|
…
|
|
linspace_op.cu
|
…
|
|
linspace_op.h
|
…
|
|
load_combine_op.cc
|
…
|
|
load_combine_op.cu
|
…
|
|
load_combine_op.h
|
…
|
|
load_op.cc
|
…
|
|
load_op.cu
|
…
|
|
load_op.h
|
…
|
|
lod_array_length_op.cc
|
…
|
|
lod_rank_table_op.cc
|
…
|
|
lod_reset_op.cc
|
…
|
|
lod_reset_op.cu
|
…
|
|
lod_reset_op.h
|
…
|
|
lod_tensor_to_array_op.cc
|
…
|
|
log_loss_op.cc
|
…
|
|
log_loss_op.cu
|
…
|
|
log_loss_op.h
|
…
|
|
lookup_sparse_table_op.cc
|
…
|
|
lookup_table_op.cc
|
…
|
|
lookup_table_op.cu
|
…
|
|
lookup_table_op.h
|
…
|
|
lrn_op.cc
|
…
|
|
lrn_op.cu
|
…
|
|
lrn_op.h
|
…
|
|
lstm_op.cc
|
…
|
|
lstm_op.cu.cc
|
…
|
|
lstm_op.h
|
…
|
|
lstm_unit_op.cc
|
…
|
|
lstm_unit_op.cu
|
…
|
|
lstm_unit_op.h
|
…
|
|
lstmp_op.cc
|
…
|
|
lstmp_op.cu
|
…
|
|
lstmp_op.h
|
…
|
|
margin_rank_loss_op.cc
|
…
|
|
margin_rank_loss_op.cu
|
…
|
|
margin_rank_loss_op.h
|
…
|
|
math.h
|
…
|
|
matmul_op.cc
|
…
|
|
max_sequence_len_op.cc
|
…
|
|
maxout_op.cc
|
…
|
|
maxout_op.cu.cc
|
…
|
|
maxout_op.h
|
…
|
|
mean_iou_op.cc
|
…
|
|
mean_iou_op.cu
|
…
|
|
mean_iou_op.h
|
…
|
|
mean_op.cc
|
…
|
|
mean_op.cu
|
…
|
|
mean_op.h
|
…
|
|
merge_lod_tensor_op.cc
|
…
|
|
merge_selected_rows_op.cc
|
…
|
|
merge_selected_rows_op.cu.cc
|
…
|
|
merge_selected_rows_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
|
…
|
|
mul_op.cc
|
…
|
|
mul_op.cu.cc
|
…
|
|
mul_op.h
|
…
|
|
multiplex_op.cc
|
…
|
|
multiplex_op.cu
|
…
|
|
multiplex_op.h
|
…
|
|
nce_op.cc
|
…
|
|
nce_op.h
|
…
|
|
norm_op.cc
|
…
|
|
norm_op.cu
|
…
|
|
norm_op.h
|
…
|
|
one_hot_op.cc
|
…
|
|
one_hot_op.cu
|
…
|
|
one_hot_op.h
|
…
|
|
pad2d_op.cc
|
…
|
|
pad2d_op.cu
|
…
|
|
pad_constant_like_op.cc
|
…
|
|
pad_constant_like_op.cu
|
…
|
|
pad_constant_like_op.h
|
…
|
|
pad_op.cc
|
…
|
|
pad_op.cu
|
…
|
|
pad_op.h
|
…
|
|
pixel_shuffle_op.cc
|
…
|
|
pixel_shuffle_op.cu
|
…
|
|
pixel_shuffle_op.h
|
…
|
|
pool_cudnn_op.cu.cc
|
…
|
|
pool_op.cc
|
…
|
|
pool_op.cu.cc
|
…
|
|
pool_op.h
|
…
|
|
pool_with_index_op.cc
|
…
|
|
pool_with_index_op.cu.cc
|
…
|
|
pool_with_index_op.h
|
…
|
|
positive_negative_pair_op.cc
|
…
|
|
positive_negative_pair_op.h
|
…
|
|
prelu_op.cc
|
…
|
|
prelu_op.cu
|
…
|
|
prelu_op.h
|
…
|
|
print_op.cc
|
…
|
|
psroi_pool_op.cc
|
…
|
|
psroi_pool_op.cu
|
…
|
|
psroi_pool_op.h
|
…
|
|
py_func_op.cc
|
…
|
|
py_func_op.h
|
…
|
|
quantize_op.cc
|
…
|
|
quantize_op.h
|
…
|
|
random_crop_op.cc
|
…
|
|
random_crop_op.cu
|
…
|
|
random_crop_op.h
|
…
|
|
range_op.cc
|
…
|
|
range_op.cu
|
…
|
|
range_op.h
|
…
|
|
rank_loss_op.cc
|
…
|
|
rank_loss_op.cu
|
…
|
|
rank_loss_op.h
|
…
|
|
recurrent_op.cc
|
…
|
|
reorder_lod_tensor_by_rank_op.cc
|
…
|
|
requantize_op.cc
|
…
|
|
requantize_op.h
|
…
|
|
reshape_op.cc
|
…
|
|
reverse_op.cc
|
…
|
|
reverse_op.cu
|
…
|
|
reverse_op.h
|
…
|
|
rnn_memory_helper_op.cc
|
…
|
|
roi_align_op.cc
|
…
|
|
roi_align_op.cu
|
…
|
|
roi_align_op.h
|
…
|
|
roi_pool_op.cc
|
…
|
|
roi_pool_op.cu
|
…
|
|
roi_pool_op.h
|
…
|
|
row_conv_op.cc
|
…
|
|
row_conv_op.cu
|
…
|
|
row_conv_op.h
|
…
|
|
sample_logits_op.cc
|
…
|
|
sample_logits_op.cu
|
…
|
|
sample_logits_op.h
|
…
|
|
sampling_id_op.cc
|
…
|
|
sampling_id_op.cu
|
…
|
|
sampling_id_op.h
|
…
|
|
save_combine_op.cc
|
…
|
|
save_combine_op.cu
|
…
|
|
save_combine_op.h
|
…
|
|
save_load_combine_op_test.cc
|
…
|
|
save_load_op_test.cc
|
…
|
|
save_op.cc
|
…
|
|
save_op.cu
|
…
|
|
save_op.h
|
fix some bug when merge sparse embedding parameters, test=develop (#18223)
|
6 years ago |
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
|
…
|
|
selu_op.cc
|
…
|
|
selu_op.cu
|
…
|
|
selu_op.h
|
…
|
|
shape_op.cc
|
…
|
|
shape_op.cu
|
…
|
|
shape_op.h
|
…
|
|
shrink_rnn_memory_op.cc
|
…
|
|
shuffle_channel_op.cc
|
…
|
|
shuffle_channel_op.cu
|
…
|
|
shuffle_channel_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
|
…
|
|
sign_op.cu
|
…
|
|
sign_op.h
|
…
|
|
similarity_focus_op.cc
|
…
|
|
similarity_focus_op.h
|
…
|
|
slice_op.cc
|
Fix slice op shape=-1 bug (#18107)
|
6 years ago |
slice_op.cu
|
…
|
|
slice_op.h
|
Fix slice op shape=-1 bug (#18107)
|
6 years ago |
smooth_l1_loss_op.cc
|
…
|
|
smooth_l1_loss_op.cu
|
…
|
|
smooth_l1_loss_op.h
|
…
|
|
softmax_cudnn_op.cu.cc
|
…
|
|
softmax_op.cc
|
…
|
|
softmax_op.cu.cc
|
…
|
|
softmax_op.h
|
…
|
|
softmax_with_cross_entropy_op.cc
|
…
|
|
softmax_with_cross_entropy_op.cu
|
…
|
|
softmax_with_cross_entropy_op.h
|
…
|
|
space_to_depth_op.cc
|
…
|
|
space_to_depth_op.cu
|
…
|
|
space_to_depth_op.h
|
…
|
|
spectral_norm_op.cc
|
…
|
|
spectral_norm_op.cu
|
…
|
|
spectral_norm_op.h
|
…
|
|
split_lod_tensor_op.cc
|
…
|
|
split_op.cc
|
…
|
|
split_op.cu.cc
|
…
|
|
split_op.h
|
…
|
|
split_selected_rows_op.cc
|
…
|
|
split_selected_rows_op.cu
|
…
|
|
split_selected_rows_op.h
|
…
|
|
spp_op.cc
|
…
|
|
spp_op.cu.cc
|
…
|
|
spp_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
|
…
|
|
squeeze_op.cc
|
…
|
|
stack_op.cc
|
…
|
|
stack_op.cu
|
…
|
|
stack_op.h
|
…
|
|
strided_memcpy.h
|
…
|
|
strided_memcpy_test.cc
|
…
|
|
sum_op.cc
|
…
|
|
sum_op.cu
|
…
|
|
sum_op.h
|
…
|
|
sync_batch_norm_op.cc
|
…
|
|
sync_batch_norm_op.cu
|
…
|
|
teacher_student_sigmoid_loss_op.cc
|
…
|
|
teacher_student_sigmoid_loss_op.h
|
…
|
|
temporal_shift_op.cc
|
…
|
|
temporal_shift_op.cu
|
…
|
|
temporal_shift_op.h
|
…
|
|
tensor_array_to_tensor_op.cc
|
…
|
|
top_k_op.cc
|
…
|
|
top_k_op.cu
|
…
|
|
top_k_op.h
|
…
|
|
transpose_op.cc
|
…
|
|
transpose_op.cu.cc
|
…
|
|
transpose_op.h
|
…
|
|
tree_conv_op.cc
|
…
|
|
tree_conv_op.cu
|
…
|
|
tree_conv_op.h
|
…
|
|
truncated_gaussian_random_op.cc
|
…
|
|
truncated_gaussian_random_op.cu
|
…
|
|
unfold_op.cc
|
…
|
|
unfold_op.cu
|
…
|
|
unfold_op.h
|
…
|
|
uniform_random_batch_size_like_op.cc
|
…
|
|
uniform_random_op.cc
|
…
|
|
uniform_random_op.cu
|
…
|
|
unpool_op.cc
|
…
|
|
unpool_op.cu.cc
|
…
|
|
unpool_op.h
|
…
|
|
unsqueeze_op.cc
|
…
|
|
unstack_op.cc
|
…
|
|
unstack_op.h
|
…
|
|
warpctc_cudnn_op.cu.cc
|
…
|
|
warpctc_op.cc
|
…
|
|
warpctc_op.cu.cc
|
…
|
|
warpctc_op.h
|
…
|
|
where_op.cc
|
…
|
|
where_op.cu
|
…
|
|
where_op.h
|
…
|
|