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.

Leo Chen 6115c14fca Pool2d cuda kernel supports fp16 (#28316) ... * pool2d cuda kernel supports fp16 * fix compile issue of template * add ut		4 years ago
..
detail	…
CMakeLists.txt	Add the cpu version of segment sum mean max min op	5 years ago
algorithm.h	…
beam_search.cc	use iwyu clean include (#27267)	5 years ago
beam_search.cu	fix the error message for the math dir	5 years ago
beam_search.h	…
beam_search_test.cc	use iwyu clean include (#27267)	5 years ago
bert_encoder_functor.cu	…
bert_encoder_functor.h	…
blas.cc	use iwyu clean include (#27267)	5 years ago
blas.h	use iwyu clean include (#27267)	5 years ago
blas_impl.cu.h	add fp16 for matmul (#27523)	5 years ago
blas_impl.h	use iwyu clean include (#27267)	5 years ago
bloomfilter.h	…
compound_functors.h	…
concat.hip.cu	…
concat_and_split.cc	use iwyu clean include (#27267)	5 years ago
concat_and_split.cu	…
concat_and_split.h	…
concat_test.cc	use iwyu clean include (#27267)	5 years ago
context_project.cc	use iwyu clean include (#27267)	5 years ago
context_project.cu	…
context_project.h	use iwyu clean include (#27267)	5 years ago
cos_sim_functor.cc	use iwyu clean include (#27267)	5 years ago
cos_sim_functor.cu	…
cos_sim_functor.h	use iwyu clean include (#27267)	5 years ago
cpu_vec.h	use iwyu clean include (#27267)	5 years ago
cpu_vec_test.cc	use iwyu clean include (#27267)	5 years ago
cross_entropy.cc	use iwyu clean include (#27267)	5 years ago
cross_entropy.cu	Enhance the error messages for files in operators/math	5 years ago
cross_entropy.h	…
depthwise_conv.cu	…
depthwise_conv.h	…
fc.cc	…
fc.cu	…
fc.h	…
functors.h	…
gru_compute.cc	use iwyu clean include (#27267)	5 years ago
gru_compute.cu	…
gru_compute.h	…
im2col.cc	use iwyu clean include (#27267)	5 years ago
im2col.cu	Enhance the error messages for files in operators/math	5 years ago
im2col.h	…
im2col_cfo_cpu.h	…
im2col_test.cc	…
lstm_compute.cc	use iwyu clean include (#27267)	5 years ago
lstm_compute.cu	…
lstm_compute.h	…
math_cuda_utils.h	…
math_function.cc	xpu support for fill_constant Op (#27675)	4 years ago
math_function.cu	enhance reduce op which can reduce tensor with arbitrary rank	5 years ago
math_function.h	xpu support for fill_constant Op (#27675)	4 years ago
math_function_impl.h	xpu support for fill_constant Op (#27675)	4 years ago
math_function_test.cc	Error description optimize for the math dir	5 years ago
math_function_test.cu	Error description optimize for math dir	5 years ago
matrix_bit_code.cc	use iwyu clean include (#27267)	5 years ago
matrix_bit_code.h	…
matrix_inverse.cc	…
matrix_inverse.cu.cc	use iwyu clean include (#27267)	5 years ago
matrix_inverse.h	…
maxouting.cc	…
maxouting.cu	…
maxouting.h	…
padding.h	Error description optimize for math dir	5 years ago
pooling.cc	fix Wmaybe-uninitialized warning in pooling.cc, test=develop (#28126)	4 years ago
pooling.cu	Pool2d cuda kernel supports fp16 (#28316)	4 years ago
pooling.h	Pool2d cuda kernel supports fp16 (#28316)	4 years ago
prelu.cu	…
prelu.h	…
sample_prob.cc	use iwyu clean include (#27267)	5 years ago
sample_prob.cu	…
sample_prob.h	use iwyu clean include (#27267)	5 years ago
sampler.cc	…
sampler.h	Error description optimize for the math dir	5 years ago
segment_pooling.cc	Add the cpu version of segment sum mean max min op	5 years ago
segment_pooling.cu	refine gpu kernel config for Paddle (#28085)	4 years ago
segment_pooling.h	Add the cpu version of segment sum mean max min op	5 years ago
selected_rows_functor.cc	update error info for selected_rows_functor	5 years ago
selected_rows_functor.cu	update error info for selected_rows_functor	5 years ago
selected_rows_functor.h	…
selected_rows_functor_test.cc	…
selected_rows_functor_test.cu.cc	use iwyu clean include (#27267)	5 years ago
sequence2batch.cc	use iwyu clean include (#27267)	5 years ago
sequence2batch.cu	…
sequence2batch.h	…
sequence_padding.cc	use iwyu clean include (#27267)	5 years ago
sequence_padding.cu	…
sequence_padding.h	…
sequence_padding_test.cc	use iwyu clean include (#27267)	5 years ago
sequence_pooling.cc	…
sequence_pooling.cu	…
sequence_pooling.h	…
sequence_pooling_test.cc	…
sequence_scale.cc	add support to float64 input of warpctc op. (#27399)	5 years ago
sequence_scale.cu	add support to float64 input of warpctc op. (#27399)	5 years ago
sequence_scale.h	use iwyu clean include (#27267)	5 years ago
softmax.cc	…
softmax.cu	…
softmax.h	…
softmax_impl.h	…
tree2col.cc	…
tree2col.cu	…
tree2col.h	…
unpooling.cc	…
unpooling.cu	…
unpooling.h	…
vol2col.cc	use iwyu clean include (#27267)	5 years ago
vol2col.cu	Error description optimize for the math dir	5 years ago
vol2col.h	…
vol2col_test.cc	use iwyu clean include (#27267)	5 years ago