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/reduce_ops
Qi Li b85c8e03be
[ROCM] fix reduce op, test=develop (#31478)
4 years ago
..
CMakeLists.txt [ROCM] update fluid operators for rocm (part2), test=develop (#31211) 4 years ago
check_reduce_rank_test.cu
cub_reduce.h [ROCM] fix dropout and remove hipcub, test=develop (#31455) 4 years ago
frobenius_norm_op.cc use iwyu clean include (#27267) 4 years ago
frobenius_norm_op.cu
frobenius_norm_op.h
logsumexp_op.cc refine logsumexp error message and docs (#27713) 4 years ago
logsumexp_op.cu
logsumexp_op.h
logsumexp_op.part.cu
reduce_all_op.cc use iwyu clean include (#27267) 4 years ago
reduce_all_op.cu
reduce_all_op.h
reduce_any_op.cc use iwyu clean include (#27267) 4 years ago
reduce_any_op.cu
reduce_any_op.h
reduce_max_op.cc
reduce_max_op.cu
reduce_max_op.part.cu
reduce_max_op_xpu.cc support reduce_max op on kunlun (#30581) 4 years ago
reduce_mean_op.cc
reduce_mean_op.cu [ROCM] fix dropout and remove hipcub, test=develop (#31455) 4 years ago
reduce_mean_op.h
reduce_mean_op.part.cu
reduce_mean_op_xpu.cc Add reduce sum and reduce mean xpu op (#27939) 4 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 4 years ago
reduce_op_function.h
reduce_op_xpu.h support reduce_max op on kunlun (#30581) 4 years ago
reduce_prod_op.cc use iwyu clean include (#27267) 4 years ago
reduce_prod_op.cu [ROCM] fix reduce op, test=develop (#31478) 4 years ago
reduce_prod_op.h
reduce_prod_op.part.cu
reduce_sum_op.cc Make transpose, trace, kron, reshape, sum op support complex type (#29321) 4 years ago
reduce_sum_op.cu [ROCM] fix dropout and remove hipcub, test=develop (#31455) 4 years ago
reduce_sum_op.h
reduce_sum_op.part.cu [Complex] Add support for complex grad accumulated (#29889) 4 years ago
reduce_sum_op_xpu.cc support reduce_max op on kunlun (#30581) 4 years ago
unity_build_rule.cmake Compiling operator libraries with Unity build (#29130) 4 years ago