Searched refs:reduction_op_ (Results 1 – 1 of 1) sorted by relevance
71 reduction_op_ = ncclMin; in NcclReduceOpBase()73 reduction_op_ = ncclMax; in NcclReduceOpBase()75 reduction_op_ = ncclSum; in NcclReduceOpBase()77 reduction_op_ = ncclProd; in NcclReduceOpBase()84 ncclRedOp_t reduction_op() const { return reduction_op_; } in reduction_op()87 ncclRedOp_t reduction_op_; member in tensorflow::__anonc894cd950111::NcclReduceOpBase190 ncclRedOp_t reduction_op_; member in tensorflow::__anonc894cd950111::NcclReduceRecvKernel