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.

Jack Zhou 63203c4abc enhance reduce op which can reduce tensor with arbitrary rank ... enhance reduce op which can reduce tensor with arbitrary rank		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	…
frobenius_norm_op.cu	…
frobenius_norm_op.h	…
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	…
reduce_all_op.cu	…
reduce_all_op.h	…
reduce_any_op.cc	…
reduce_any_op.cu	…
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	enhance reduce op which can reduce tensor with arbitrary rank	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	…
reduce_sum_op.h	…
reduce_sum_op.part.cu	…