You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Paddle/paddle/fluid/operators
Zeng Jinle 0b49e43d3a
Merge pull request #16144 from sneaxiy/rnn_mem_opt
6 years ago
..
benchmark Enhance the op benchmark: (#16066) 6 years ago
controlflow fix travis-ci format check 6 years ago
csp Refine operator cmake (#14413) 6 years ago
detail Merge pull request #14933 from sneaxiy/rewrite_ddim 6 years ago
detection resolve #15618 (#16114) 6 years ago
distributed Profiler refine and add CUDA runtime api tracer (#15301) 6 years ago
distributed_ops Merge pull request #15536 from jacquesqiao/fix-prefetch-one-parameter 6 years ago
elementwise simplify the jitkernel templates and tests 6 years ago
fused refine with comments 6 years ago
jit enhance jitkernel unit test 6 years ago
math Merge pull request #16144 from sneaxiy/rnn_mem_opt 6 years ago
metrics Fp16 training (#14992) 6 years ago
mkldnn Add Requantize OP (#15318) 6 years ago
nccl Revert the changes of VLOG 6 years ago
ngraph remove const_cast and refactor ngraph engine code (#15925) 6 years ago
optimizers simplify the jitkernel templates and tests 6 years ago
reader Revert "Revert "Add Event for TensorCopy"" (#16035) 6 years ago
reduce_ops test=develop 6 years ago
sequence_ops Merge pull request #16144 from sneaxiy/rnn_mem_opt 6 years ago
tensorrt fix comments and fix cpplint 6 years ago
CMakeLists.txt Merge pull request #15609 from xuezhong/add_sample_logits_op 6 years ago
activation_cudnn.cu.cc polish cudnn related code and fix bug. (#15164) 6 years ago
activation_cudnn_op.cu.cc polish cudnn related code and fix bug. (#15164) 6 years ago
activation_op.cc refine doc 6 years ago
activation_op.cu Add fp16 backward support (#14202) 6 years ago
activation_op.h Add the inverse trigonometric function 6 years ago
add_position_encoding_op.cc position encoding && log loss 6 years ago
add_position_encoding_op.h Exhaustive search for cuDNN conv. (#14286) 6 years ago
affine_channel_op.cc Add affine channel op to speed and save memory for faster-rcnn model. (#13919) 6 years ago
affine_channel_op.cu Add generate_mask_labels_op to support Mask-RCNN and refine some code. (#15371) 6 years ago
affine_grid_cudnn_op.cu.cc Add affine grid generator op (#12238) 6 years ago
affine_grid_op.cc polish code 6 years ago
affine_grid_op.h Fix build error of affine grid op in mac os. (#14237) 6 years ago
alloc_continuous_space_op.cc Add alloc_continuous_space_op (#15900) 6 years ago
arg_max_op.cc Change tensor uses proto::VarType::type 6 years ago
arg_max_op.cu Change tensor uses proto::VarType::type 6 years ago
arg_min_max_op_base.h Polish arg_min_max_op 7 years ago
arg_min_op.cc Change tensor uses proto::VarType::type 6 years ago
arg_min_op.cu Change tensor uses proto::VarType::type 6 years ago
argsort_op.cc Set the right shape of selected_rows (#13723) 6 years ago
argsort_op.cu Enhance cuda code & unittest for argsort_op 7 years ago
argsort_op.h Simplify the computation in cpu 7 years ago
array_operator.h Revert the changes of VLOG 6 years ago
array_to_lod_tensor_op.cc Change tensor uses proto::VarType::type 6 years ago
assign_op.cc
assign_value_op.cc Register assign_value_op an empty grad_op 7 years ago
assign_value_op.cu.cc Revert ""cherry picked operators changes" (#12184)" (#12747) 7 years ago
assign_value_op.h
attention_lstm_op.cc fix warnings (#15790) 6 years ago
attention_lstm_op.h implement attention lstm cpu forward 7 years ago
average_accumulates_op.cc Change tensor uses proto::VarType::type 6 years ago
average_accumulates_op.cu
average_accumulates_op.h small AverageOptimizer enhance. (#11761) 7 years ago
batch_norm_op.cc Merge remote-tracking branch 'origin/develop' into feature/ir_inplace_pass 6 years ago
batch_norm_op.cu Enable BatchNorm to use global mean and variane during training (#14630) 6 years ago
batch_norm_op.h add url of cuda9.0_cudnn7_avx_mkl library 7 years ago
batch_size_like.h Simplize API Reference Documentation 7 years ago
beam_search_decode_op.cc add per kernel config and remove const_cast. 6 years ago
beam_search_decode_op.h Change *(smart_ptr.get()) -> *smart_ptr 6 years ago
beam_search_decode_op_test.cc Make python unit test of beam_search_op and beam_searc_decode_op run correctly 7 years ago
beam_search_op.cc Make parent_idx a dispensable output for beam_search op to support models saved by older paddle version. (#16106) 6 years ago
beam_search_op.cu.cc Add the CUDA kernel for beam_search op (#15020) 6 years ago
beam_search_op.h Make parent_idx a dispensable output for beam_search op to support models saved by older paddle version. (#16106) 6 years ago
bilinear_tensor_product_op.cc
bilinear_tensor_product_op.cu Fix Eigen macro when using GPU 6 years ago
bilinear_tensor_product_op.h Optimize bilinear tensor product op (#14485) 6 years ago
bpr_loss_op.cc Change tensor uses proto::VarType::type 6 years ago
bpr_loss_op.h Add the CUDA kernel for beam_search op (#15020) 6 years ago
cast_op.cc polish the cast op doc (#16078) 6 years ago
cast_op.cu
cast_op.h Revert "cherry picked windows patches." 6 years ago
chunk_eval_op.cc update by comment 7 years ago
chunk_eval_op.h
clip_by_norm_op.cc fix typo 7 years ago
clip_by_norm_op.cu
clip_by_norm_op.h rewrite variable type 6 years ago
clip_op.cc
clip_op.cu
clip_op.h fix sparse gradient clip 6 years ago
concat_op.cc Include MKL-DNN header to concat op only when flag is set 6 years ago
concat_op.cu.cc enable more type for splitOp and ConcatOp 7 years ago
concat_op.h Refine Split op (#13967) 6 years ago
conv_cudnn_op.cu.cc add per kernel config and remove const_cast. 6 years ago
conv_cudnn_op_cache.h add per kernel config and remove const_cast. 6 years ago
conv_fusion_op.cc Inception fusion operator. (#14968) 6 years ago
conv_fusion_op.cu.cc polish 6 years ago
conv_op.cc Enable function coverage for U8/S8 ConvMKLDNNOpKernel 6 years ago
conv_op.cu.cc
conv_op.h Memory optimization of depthwise conv op and group norm op (#15313) 6 years ago
conv_shift_op.cc Set the right shape of selected_rows (#13723) 6 years ago
conv_shift_op.cu
conv_shift_op.h
conv_transpose_cudnn_op.cu.cc Revert conv transpose cudnn (#15514) 6 years ago
conv_transpose_op.cc MKLDNN: Add UT for conv_transpose_mkldnn op. (#16030) 6 years ago
conv_transpose_op.cu.cc Speed depthwise transposed conv2d. (#11740) 7 years ago
conv_transpose_op.h Optimization of Kernels that related to DeepLabv3+ (#13534) 6 years ago
cos_sim_op.cc update by comment 7 years ago
cos_sim_op.cu Fix Eigen macro when using GPU 6 years ago
cos_sim_op.h
crf_decoding_op.cc fix warnings (#15790) 6 years ago
crf_decoding_op.h simplify the jitkernel templates and tests 6 years ago
crop_op.cc Change tensor uses proto::VarType::type 6 years ago
crop_op.cu Fix Eigen macro when using GPU 6 years ago
crop_op.h rewrite ddim 6 years ago
cross_entropy_op.cc fix travis-ci 6 years ago
cross_entropy_op.cu combine op files 6 years ago
cross_entropy_op.h combine op files 6 years ago
ctc_align_op.cc Change tensor uses proto::VarType::type 6 years ago
ctc_align_op.cu
ctc_align_op.h
cudnn_lstm_op.cc code refine 6 years ago
cudnn_lstm_op.cu.cc merge develop 6 years ago
cudnn_rnn_cache.h rewrite variable type 6 years ago
cum_op.h fix test issues on windows 6 years ago
cumsum_op.cc Polish comsum, DynamicRNN 7 years ago
cumsum_op.cu
data_norm_op.cc remove mkldnn & fix commit 6 years ago
data_norm_op.h data_norm 6 years ago
delete_var_op.cc [1.1] Load vars on PSERVER (#14037) 6 years ago
dequantize_op.cc Fix comments misunderstanding 6 years ago
dequantize_op.h Add Dequantize OP 6 years ago
detection_map_op.cc Change tensor uses proto::VarType::type 6 years ago
detection_map_op.h Revert "Revert "Merge pull request #13431 from chengduoZH/refine_lod"" 6 years ago
dropout_op.cc add ShareLoD for dropout_grad (#14616) 6 years ago
dropout_op.cu Some improvements to support bert mixed precision training (#15585) 6 years ago
dropout_op.h modify dropout att; test=develop 6 years ago
dropout_op_test.cc minor fix 6 years ago
edit_distance_op.cc
edit_distance_op.cu
edit_distance_op.h
expand_op.cc refine code 6 years ago
expand_op.cu support multiple var types for expand op, test=develop 6 years ago
expand_op.h rewrite ddim 6 years ago
fake_dequantize_op.cc update the input format of channel wise dequantize op. 6 years ago
fake_dequantize_op.cu add channel wise dequantize op. 6 years ago
fake_dequantize_op.h Reduce redundant code for channel wise dequant op. test=develop 6 years ago
fake_quantize_op.cc add channel wise dequantize op. 6 years ago
fake_quantize_op.cu add channel wise quantize op. 6 years ago
fake_quantize_op.h add channel wise quantize op. 6 years ago
fc_op.cc fix warnings (#15790) 6 years ago
fc_op.h
fill_constant_batch_size_like_op.cc Simplize API Reference Documentation 7 years ago
fill_constant_batch_size_like_op.cu.cc
fill_constant_batch_size_like_op.h
fill_constant_op.cc register float16 6 years ago
fill_constant_op.cu.cc register float16 6 years ago
fill_constant_op.h make fill_constant kernel-based 6 years ago
fill_op.cc Change tensor uses proto::VarType::type 6 years ago
fill_zeros_like_op.cc Revert "Extend fill_zeros_like_op for zero-filling an LoDTensorArray (#11496)" 7 years ago
fill_zeros_like_op.cu.cc Fp16 training (#14992) 6 years ago
fill_zeros_like_op.h Revert "Extend fill_zeros_like_op for zero-filling an LoDTensorArray (#11496)" 7 years ago
flatten_op.cc squash commits. test=develop 6 years ago
gather.cu.h Fix gather & stack op (#14355) 6 years ago
gather.h Fix gather & stack op (#14355) 6 years ago
gather_op.cc Add generate_mask_labels_op to support Mask-RCNN and refine some code. (#15371) 6 years ago
gather_op.cu Some improvements to support bert mixed precision training (#15585) 6 years ago
gather_op.h Return parent_idx in beam_search op (#15520) 6 years ago
gather_test.cc fix some bugs introduced by unfreed memory 7 years ago
gaussian_random_batch_size_like_op.cc Update some doc about API reference. (#11495) 7 years ago
gaussian_random_op.cc clean code test=develop 6 years ago
gaussian_random_op.cu Revert ""cherry picked operators changes" (#12184)" (#12747) 7 years ago
get_tensor_from_selected_rows_op.cc Fix clip.py (#14718) 6 years ago
grid_sampler_cudnn_op.cu.cc fix some inappropriate expressions in api doc for grid_sampler. test=develop 6 years ago
grid_sampler_op.cc fix grid_sampler PADDLE_ENFORCE error. test=develop (#15542) 6 years ago
grid_sampler_op.h code style fix 6 years ago
group_norm_op.cc inplace group_norm (#15754) 6 years ago
group_norm_op.cu fix pr 15313 6 years ago
group_norm_op.h Memory optimization of depthwise conv op and group norm op (#15313) 6 years ago
gru_op.cc update gru op forward kernel 6 years ago
gru_op.cu.cc update gru op forward kernel 6 years ago
gru_op.h update gru op forward kernel 6 years ago
gru_unit_op.cc change interface and api spec for dynamic_gru test=develop 6 years ago
gru_unit_op.cu Fix Eigen macro when using GPU 6 years ago
gru_unit_op.h complete gru_unite_op and test 6 years ago
hash_op.cc refine with comments 6 years ago
hash_op.h refine infershape of sequence_enumerate, hash and fuse_emb_seq_pool 6 years ago
hierarchical_sigmoid_op.cc merge reyoung optimization 6 years ago
hierarchical_sigmoid_op.h test=develop, fix debug mode unitest, hsigmoid (#15574) 6 years ago
hinge_loss_op.cc rewrite ddim 6 years ago
hinge_loss_op.cu Fix Eigen macro when using GPU 6 years ago
hinge_loss_op.h
huber_loss_op.cc Add python interface for huber regression loss 6 years ago
huber_loss_op.cu Fix Eigen macro when using GPU 6 years ago
huber_loss_op.h fix the huber loss compile issue on windows test=develop 6 years ago
im2sequence_op.cc Fix infershape of im2sequence. (#12183) 7 years ago
im2sequence_op.cu Fix Eigen macro when using GPU 6 years ago
im2sequence_op.h Fix infershape of im2sequence. (#12183) 7 years ago
increment_op.cc
increment_op.cu
increment_op.h
interpolate_op.cc refine image_resize annotation (#15976) 6 years ago
interpolate_op.cu test=develop 6 years ago
interpolate_op.h test=develop 6 years ago
is_empty_op.cc Rewrite is_empty op to avoid unnecessary data transform. (#15509) 6 years ago
is_empty_op.cu.cc Rewrite is_empty op to avoid unnecessary data transform. (#15509) 6 years ago
is_empty_op.h Rewrite is_empty op to avoid unnecessary data transform. (#15509) 6 years ago
isfinite_op.cc Change tensor uses proto::VarType::type 6 years ago
isfinite_op.cu Fix Eigen macro when using GPU 6 years ago
isfinite_op.h enhance isinf/isnan in tensor util, avoid copy back to cpu (#12688) 6 years ago
l1_norm_op.cc
l1_norm_op.cu Fix Eigen macro when using GPU 6 years ago
l1_norm_op.h
label_smooth_op.cc test=develop 6 years ago
label_smooth_op.cu
label_smooth_op.h fix windows compile (#13147) 7 years ago
layer_norm_op.cc fix warnings (#15790) 6 years ago
layer_norm_op.cu Use double to reduce 7 years ago
layer_norm_op.h simplify the jitkernel templates and tests 6 years ago
linear_chain_crf_op.cc fix warnings (#15790) 6 years ago
linear_chain_crf_op.cu
linear_chain_crf_op.h
load_combine_op.cc More restrict check load_combine_op. (#15479) 6 years ago
load_op.cc fix save and load ops on windows test=develop 6 years ago
lod_array_length_op.cc
lod_rank_table_op.cc Revert the changes of VLOG 6 years ago
lod_reset_op.cc Change tensor uses proto::VarType::type 6 years ago
lod_reset_op.cu
lod_reset_op.h
lod_tensor_to_array_op.cc Change tensor uses proto::VarType::type 6 years ago
log_loss_op.cc rewrite ddim 6 years ago
log_loss_op.cu Fix Eigen macro when using GPU 6 years ago
log_loss_op.h
lookup_sparse_table_op.cc Change tensor uses proto::VarType::type 6 years ago
lookup_table_op.cc Merge branch 'develop' of https://github.com/PaddlePaddle/Paddle into refactor-prefetch 6 years ago
lookup_table_op.cu Some improvements to support bert mixed precision training (#15585) 6 years ago
lookup_table_op.h fix problem test=develop 6 years ago
lrn_op.cc Change tensor uses proto::VarType::type 6 years ago
lrn_op.cu
lrn_op.h refine lrn_op cpu forward and speedup 6 years ago
lstm_op.cc Change tensor uses proto::VarType::type 6 years ago
lstm_op.cu.cc
lstm_op.h Merge branch 'develop' of https://github.com/PaddlePaddle/Paddle into fix_bug_for_lstmp 6 years ago
lstm_unit_op.cc
lstm_unit_op.cu
lstm_unit_op.h Revert "cherry picked windows patches." 6 years ago
lstmp_op.cc add cell clip and proj clip, fix bug for h0 6 years ago
lstmp_op.cu
lstmp_op.h Merge branch 'develop' of https://github.com/PaddlePaddle/Paddle into fix_bug_for_lstmp 6 years ago
margin_rank_loss_op.cc
margin_rank_loss_op.cu
margin_rank_loss_op.h
math.h refine code 6 years ago
matmul_op.cc modification 6 years ago
max_sequence_len_op.cc Simplize API Reference Documentation 7 years ago
maxout_op.cc modified error info for maxout op 7 years ago
maxout_op.cu.cc
maxout_op.h
mean_iou_op.cc Change tensor uses proto::VarType::type 6 years ago
mean_iou_op.cu [Feature] Add Temporary Allocator (#14875) 6 years ago
mean_iou_op.h Add mean IOU op. (#10519) 7 years ago
mean_op.cc Change tensor uses proto::VarType::type 6 years ago
mean_op.cu Fix Eigen macro when using GPU 6 years ago
mean_op.h Add fp16 backward support (#14202) 6 years ago
merge_lod_tensor_op.cc Change tensor uses proto::VarType::type 6 years ago
merge_selected_rows_op.cc Refine merge_selected_rows Doc (#14748) 6 years ago
merge_selected_rows_op.cu.cc Fix clip.py (#14718) 6 years ago
merge_selected_rows_op.h Fix clip.py (#14718) 6 years ago
minus_op.cc
minus_op.cu
minus_op.h
modified_huber_loss_op.cc rewrite ddim 6 years ago
modified_huber_loss_op.cu
modified_huber_loss_op.h
mul_op.cc merge develop 6 years ago
mul_op.cu.cc Add fp16 backward support (#14202) 6 years ago
mul_op.h Process elemwise grad op's lod. mul_op's lod 7 years ago
multiplex_op.cc Change tensor uses proto::VarType::type 6 years ago
multiplex_op.cu
multiplex_op.h
nce_op.cc Merge pull request #14893 from JiabinYang/feature/add_prefech_hs 6 years ago
nce_op.h nce add check sample lables, test=develop (#15463) 6 years ago
norm_op.cc Make the normalization operator more general and fix bug in l2_normalize. (#11348) 7 years ago
norm_op.cu Implement norm_op by CUDA instead of Eigen. (#13273) 7 years ago
norm_op.h test=develop 6 years ago
one_hot_op.cc
one_hot_op.cu Feature/template (#13093) 7 years ago
one_hot_op.h Feature/template (#13093) 7 years ago
pad2d_op.cc Change tensor uses proto::VarType::type 6 years ago
pad2d_op.cu Make pad2d support for variable paddings. (#14667) 6 years ago
pad_constant_like_op.cc Change tensor uses proto::VarType::type 6 years ago
pad_constant_like_op.cu Fix Eigen macro when using GPU 6 years ago
pad_constant_like_op.h Add pad_constant_like_op (#12943) 7 years ago
pad_op.cc
pad_op.cu Fix Eigen macro when using GPU 6 years ago
pad_op.h Add pad_constant_like_op (#12943) 7 years ago
pool_cudnn_op.cu.cc Add fp16 backward support (#14202) 6 years ago
pool_op.cc use kernel size in global_pooling. test=develop 6 years ago
pool_op.cu.cc
pool_op.h add adaptive pool 2d & 3d. test=develop 6 years ago
pool_with_index_op.cc Merge branch 'develop' of https://github.com/PaddlePaddle/Paddle into feature/tensor_type 6 years ago
pool_with_index_op.cu.cc
pool_with_index_op.h add adaptive pool 2d & 3d. test=develop 6 years ago
positive_negative_pair_op.cc Change tensor uses proto::VarType::type 6 years ago
positive_negative_pair_op.h update CMakeLists.txt 6 years ago
prelu_op.cc clean 6 years ago
prelu_op.cu add prelu gpu inference 6 years ago
prelu_op.h Refine prelu_op 6 years ago
print_op.cc Change tensor uses proto::VarType::type 6 years ago
psroi_pool_op.cc Fix merge bug 6 years ago
psroi_pool_op.cu this is for psroi_pool op, test=develop (#14796) 6 years ago
psroi_pool_op.h rewrite ddim 6 years ago
py_func_op.cc try fix py2 6 years ago
py_func_op.h try fix py2 6 years ago
quantize_op.cc Add Quantize OP 6 years ago
quantize_op.h Add Quantize OP 6 years ago
random_crop_op.cc Change tensor uses proto::VarType::type 6 years ago
random_crop_op.cu
random_crop_op.h fix security issue 27, 38 test=develop 6 years ago
rank_loss_op.cc
rank_loss_op.cu
rank_loss_op.h
recurrent_op.cc Merge pull request #16063 from sneaxiy/enhance_gc 6 years ago
reorder_lod_tensor_by_rank_op.cc refine tensor_array_write_read (#14643) 6 years ago
requantize_op.cc Add Requantize OP (#15318) 6 years ago
requantize_op.h Add Requantize OP (#15318) 6 years ago
reshape_op.cc Fix reshape bug (#16069) 6 years ago
reverse_op.cc Refine code 7 years ago
reverse_op.cu Refine code 7 years ago
reverse_op.h fix a bug 7 years ago
rnn_memory_helper_op.cc Change tensor uses proto::VarType::type 6 years ago
roi_align_op.cc Change tensor uses proto::VarType::type 6 years ago
roi_align_op.cu Add generate_mask_labels_op to support Mask-RCNN and refine some code. (#15371) 6 years ago
roi_align_op.h test=develop 6 years ago
roi_pool_op.cc Change tensor uses proto::VarType::type 6 years ago
roi_pool_op.cu Add generate_mask_labels_op to support Mask-RCNN and refine some code. (#15371) 6 years ago
roi_pool_op.h Refine and fix some code for faster-rcnn. (#13135) 7 years ago
row_conv_op.cc Fix row_conv doc 6 years ago
row_conv_op.cu
row_conv_op.h
sample_logits_op.cc remove non-ascii charactor 6 years ago
sample_logits_op.cu refine code 6 years ago
sample_logits_op.h refine code 6 years ago
sampling_id_op.cc fix 6 years ago
sampling_id_op.cu merge cpu and gpu 7 years ago
sampling_id_op.h refine 7 years ago
save_combine_op.cc fix save and load ops on windows test=develop 6 years ago
save_load_combine_op_test.cc
save_load_op_test.cc unittest case fix 7 years ago
save_op.cc fix save and load ops on windows test=develop 6 years ago
scale_op.cc squash commits. test=develop 6 years ago
scale_op.cu Add fp16 backward support (#14202) 6 years ago
scale_op.h Fix input<tensor> (#14208) 6 years ago
scatter.cu.h Fix gather & stack op (#14355) 6 years ago
scatter.h Fix gather & stack op (#14355) 6 years ago
scatter_op.cc Change tensor uses proto::VarType::type 6 years ago
scatter_op.cu
scatter_op.h Fix scatter_op python API (#12742) 7 years ago
scatter_test.cc Fix bug in uts 6 years ago
selu_op.cc Add selu (#14415) 6 years ago
selu_op.cu Add selu (#14415) 6 years ago
selu_op.h refine code 6 years ago
shape_op.cc fix shape api doc 6 years ago
shape_op.cu Add flatten op interface and enhance APIs about detection to support variable-length image. (#12422) 7 years ago
shape_op.h Add flatten op interface and enhance APIs about detection to support variable-length image. (#12422) 7 years ago
shrink_rnn_memory_op.cc refine tensor_array_write_read (#14643) 6 years ago
shuffle_channel_op.cc rewrite the comments, test=develop 6 years ago
shuffle_channel_op.cu update API.spec, test=develop 6 years ago
shuffle_channel_op.h Fix, test=develop 6 years ago
sigmoid_cross_entropy_with_logits_op.cc Add generate_mask_labels_op to support Mask-RCNN and refine some code. (#15371) 6 years ago
sigmoid_cross_entropy_with_logits_op.cu refine code 6 years ago
sigmoid_cross_entropy_with_logits_op.h Add generate_mask_labels_op to support Mask-RCNN and refine some code. (#15371) 6 years ago
sign_op.cc "fix op. test=develop" 6 years ago
sign_op.cu "fix op. test=develop" 6 years ago
sign_op.h
similarity_focus_op.cc Change tensor uses proto::VarType::type 6 years ago
similarity_focus_op.h add similarity_focus op 6 years ago
slice_op.cc add lod for slice op, test=develop 6 years ago
slice_op.cu Implement slice grad operator. #8130 (#12330) 7 years ago
slice_op.h Implement slice grad operator. #8130 (#12330) 7 years ago
smooth_l1_loss_op.cc
smooth_l1_loss_op.cu Fix Eigen macro when using GPU 6 years ago
smooth_l1_loss_op.h
softmax_cudnn_op.cu.cc Add fp16 backward support (#14202) 6 years ago
softmax_op.cc squash commits. test=develop 6 years ago
softmax_op.cu.cc Add fp16 backward support (#14202) 6 years ago
softmax_op.h - ASUM MKL integration 6 years ago
softmax_with_cross_entropy_op.cc change default option related to softmax, test=develop 6 years ago
softmax_with_cross_entropy_op.cu [Feature] support mix precision training for resnet (#14899) 6 years ago
softmax_with_cross_entropy_op.h - Softmax for Inference is enabled when ON_INFER is set 6 years ago
space_to_depth_op.cc Windows/online (#14474) 6 years ago
space_to_depth_op.cu test=develop 6 years ago
space_to_depth_op.h test=develop 6 years ago
spectral_norm_op.cc fix statement. test=develop 6 years ago
spectral_norm_op.cu fix spectral_norm doc. test=develop 6 years ago
spectral_norm_op.h fix format. test=develop 6 years ago
split_lod_tensor_op.cc fix bug in if-else op, test=develop 6 years ago
split_op.cc Refine Split op (#13967) 6 years ago
split_op.cu.cc enable more type for splitOp and ConcatOp 7 years ago
split_op.h Refine Split op (#13967) 6 years ago
split_selected_rows_op.cc update height_sections to int64_t 6 years ago
split_selected_rows_op.cu
split_selected_rows_op.h split selected rows op should always init output selected rows 6 years ago
spp_op.cc
spp_op.cu.cc
spp_op.h add adaptive pool 2d & 3d. test=develop 6 years ago
squared_l2_distance_op.cc
squared_l2_distance_op.cu Fix Eigen macro when using GPU 6 years ago
squared_l2_distance_op.h
squared_l2_norm_op.cc
squared_l2_norm_op.cu Fix Eigen macro when using GPU 6 years ago
squared_l2_norm_op.h
squeeze_op.cc Refine reshape_grad and transpose_grad (#13074) 7 years ago
stack_op.cc Fix gather & stack op (#14355) 6 years ago
stack_op.cu Some improvements to support bert mixed precision training (#15585) 6 years ago
stack_op.h Add the macro for NVCC (test=develop) 6 years ago
strided_memcpy.h rewrite ddim 6 years ago
strided_memcpy_test.cc refactor(memory): rewrite memory allocation and make it extentable 6 years ago
sum_op.cc fix sum_op selected rows test=develop 6 years ago
sum_op.cu Fix Eigen macro when using GPU 6 years ago
sum_op.h rewrite variable type 6 years ago
teacher_student_sigmoid_loss_op.cc remove mkl & fix commit 6 years ago
teacher_student_sigmoid_loss_op.h remove some comments & refine doc & put template class in .h 6 years ago
tensor_array_to_tensor_op.cc Use sub scope in tensor_array_to_tensor op. (#14524) 6 years ago
top_k_op.cc Make topk op support variable k. (#15044) 6 years ago
top_k_op.cu Make topk op support variable k. (#15044) 6 years ago
top_k_op.h Make topk op support variable k. (#15044) 6 years ago
transpose_op.cc - Added transpose/transpose2 MKLDNN grad ops 6 years ago
transpose_op.cu.cc Some improvements to support bert mixed precision training (#15585) 6 years ago
transpose_op.h
tree_conv_op.cc Tree conv op (#15217) 6 years ago
tree_conv_op.cu Tree conv op (#15217) 6 years ago
tree_conv_op.h Tree conv op (#15217) 6 years ago
truncated_gaussian_random_op.cc Fix truncated norm (#13785) 6 years ago
truncated_gaussian_random_op.cu Fix truncated norm (#13785) 6 years ago
uniform_random_batch_size_like_op.cc "fix based comments" 7 years ago
uniform_random_op.cc shape type to int64_t, test=develop 6 years ago
uniform_random_op.cu fix shape type in uniform_random_op.cu 6 years ago
unpool_op.cc Change tensor uses proto::VarType::type 6 years ago
unpool_op.cu.cc
unpool_op.h
unsqueeze_op.cc Refine reshape_grad and transpose_grad (#13074) 7 years ago
unstack_op.cc add unstack_op 7 years ago
unstack_op.h add unstack_op 7 years ago
warpctc_cudnn_op.cu.cc Revert conv transpose cudnn (#15514) 6 years ago
warpctc_op.cc polish code 6 years ago
warpctc_op.cu.cc
warpctc_op.h Complete sequence_padding GPU kernel 7 years ago