Searched defs:all_reduce (Results 1 – 4 of 4) sorted by relevance
| /third_party/boost/boost/mpi/collectives/ |
| D | all_reduce.hpp | 90 all_reduce(const communicator& comm, const T* in_values, int n, T* out_values, in all_reduce() function
99 all_reduce(const communicator& comm, inplace_t<T*> inout_values, int n, Op op) in all_reduce() function
106 all_reduce(const communicator& comm, inplace_t<T> inout_values, Op op) in all_reduce() function
113 all_reduce(const communicator& comm, const T& in_value, T& out_value, Op op) in all_reduce() function
120 T all_reduce(const communicator& comm, const T& in_value, Op op) in all_reduce() function
|
| /third_party/boost/boost/graph/parallel/ |
| D | algorithm.hpp | 55 all_reduce(ProcessGroup pg, const T& value, BinaryOperation bin_op) in all_reduce() function
|
| /third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pre_activate/ |
| D | insert_tensor_move_for_hccl_op.py | 21 all_reduce = P.AllReduce() variable
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | arithmetic_simplify.cc | 164 AnfNodePtr all_reduce = NewCNode({all_reduce_prim, add}, fg); in operator ()() local
|