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.

zhupengyang cc3306f7c8 restruct logsumexp to speed up compiling (#27191)		5 years ago
..
CMakeLists.txt	[cuda11 support] change the CMakeLists to support the cuda11 (#27124)	5 years ago
check_reduce_rank_test.cu	…
cub_reduce.h	OP error message enhancement of l2_normalize, matmul, mean, etc	5 years ago
frobenius_norm_op.cc	Add new norm api, support frobenius norm and p-order vector norm. (#23716)	5 years ago
frobenius_norm_op.cu	Add new norm api, support frobenius norm and p-order vector norm. (#23716)	5 years ago
frobenius_norm_op.h	Add new norm api, support frobenius norm and p-order vector norm. (#23716)	5 years ago
logsumexp_op.cc	restruct logsumexp to speed up compiling (#27191)	5 years ago
logsumexp_op.cu	restruct logsumexp to speed up compiling (#27191)	5 years ago
logsumexp_op.h	restruct logsumexp to speed up compiling (#27191)	5 years ago
logsumexp_op.part.cu	restruct logsumexp to speed up compiling (#27191)	5 years ago
reduce_all_op.cc	Add the sum op to API 2.0， add some parameters for new api	5 years ago
reduce_all_op.cu	Add the sum op to API 2.0， add some parameters for new api	5 years ago
reduce_all_op.h	…
reduce_any_op.cc	Add the sum op to API 2.0， add some parameters for new api	5 years ago
reduce_any_op.cu	Add the sum op to API 2.0， add some parameters for new api	5 years ago
reduce_any_op.h	…
reduce_max_op.cc	…
reduce_max_op.cu	…
reduce_max_op.part.cu	…
reduce_mean_op.cc	mean: not support int32, int64; add check for axis (#26401)	5 years ago
reduce_mean_op.cu	mean: not support int32, int64; add check for axis (#26401)	5 years ago
reduce_mean_op.h	…
reduce_mean_op.part.cu	reduce_mean error if keepdim=True and reduce_all=True (#26614)	5 years ago
reduce_min_max_op.h	…
reduce_min_op.cc	…
reduce_min_op.cu	…
reduce_min_op.part.cu	…
reduce_op.h	reduce_mean error if keepdim=True and reduce_all=True (#26614)	5 years ago
reduce_op_function.h	…
reduce_prod_op.cc	…
reduce_prod_op.cu	…
reduce_prod_op.h	…
reduce_prod_op.part.cu	…
reduce_sum_op.cc	Add double grad in reduce sum (#27115)	5 years ago
reduce_sum_op.cu	Add the sum op to API 2.0， add some parameters for new api	5 years ago
reduce_sum_op.h	Add the sum op to API 2.0， add some parameters for new api	5 years ago
reduce_sum_op.part.cu	…