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 b67ded04f2 Support gradient accumulation of fp16 in imperative mode (#24823) ... * support gradient accumulation of fp16 in imperative mode, test=develop * enhance coverage test, test=develop * follow comments, test=develop		6 years ago
..
detail	…
CMakeLists.txt	…
algorithm.h	…
beam_search.cc	…
beam_search.cu	…
beam_search.h	…
beam_search_test.cc	…
bert_encoder_functor.cu	…
bert_encoder_functor.h	…
blas.cc	…
blas.h	…
blas_impl.cu.h	…
blas_impl.h	…
bloomfilter.h	…
compound_functors.h	…
concat.hip.cu	…
concat_and_split.cc	Add macro BOOST_GET to enrich the error information of boost :: get (#24175)	6 years ago
concat_and_split.cu	Add macro BOOST_GET to enrich the error information of boost :: get (#24175)	6 years ago
concat_and_split.h	…
concat_test.cc	…
context_project.cc	…
context_project.cu	…
context_project.h	…
cos_sim_functor.cc	…
cos_sim_functor.cu	…
cos_sim_functor.h	…
cpu_vec.h	…
cpu_vec_test.cc	…
cross_entropy.cc	…
cross_entropy.cu	…
cross_entropy.h	…
depthwise_conv.cu	…
depthwise_conv.h	…
fc.cc	…
fc.cu	…
fc.h	…
functors.h	…
gru_compute.cc	…
gru_compute.cu	…
gru_compute.h	…
im2col.cc	…
im2col.cu	…
im2col.h	…
im2col_cfo_cpu.h	…
im2col_test.cc	…
lstm_compute.cc	…
lstm_compute.cu	…
lstm_compute.h	…
math_cuda_utils.h	…
math_function.cc	Support gradient accumulation of fp16 in imperative mode (#24823)	6 years ago
math_function.cu	Support gradient accumulation of fp16 in imperative mode (#24823)	6 years ago
math_function.h	Support gradient accumulation of fp16 in imperative mode (#24823)	6 years ago
math_function_impl.h	…
math_function_test.cc	…
math_function_test.cu	…
matrix_bit_code.cc	…
matrix_bit_code.h	…
matrix_inverse.cc	…
matrix_inverse.cu.cc	…
matrix_inverse.h	…
maxouting.cc	…
maxouting.cu	…
maxouting.h	…
padding.h	…
pooling.cc	…
pooling.cu	…
pooling.h	…
prelu.cu	…
prelu.h	…
sample_prob.cc	…
sample_prob.cu	…
sample_prob.h	…
sampler.cc	…
sampler.h	…
selected_rows_functor.cc	Add macro BOOST_GET to enrich the error information of boost :: get (#24175)	6 years ago
selected_rows_functor.cu	Add macro BOOST_GET to enrich the error information of boost :: get (#24175)	6 years ago
selected_rows_functor.h	…
selected_rows_functor_test.cc	…
selected_rows_functor_test.cu.cc	…
sequence2batch.cc	…
sequence2batch.cu	…
sequence2batch.h	…
sequence_padding.cc	…
sequence_padding.cu	…
sequence_padding.h	…
sequence_padding_test.cc	…
sequence_pooling.cc	…
sequence_pooling.cu	…
sequence_pooling.h	…
sequence_pooling_test.cc	…
sequence_scale.cc	…
sequence_scale.cu	…
sequence_scale.h	…
softmax.cc	…
softmax.cu	…
softmax.h	…
softmax_impl.h	…
tree2col.cc	Add macro BOOST_GET to enrich the error information of boost :: get (#24175)	6 years ago
tree2col.cu	Add macro BOOST_GET to enrich the error information of boost :: get (#24175)	6 years ago
tree2col.h	…
unpooling.cc	…
unpooling.cu	…
unpooling.h	…
vol2col.cc	…
vol2col.cu	…
vol2col.h	…
vol2col_test.cc	…