Searched defs:reduction_op (Results 1 – 7 of 7) sorted by relevance
61 Status ReductionOp(const string& merge_op, ncclRedOp_t* reduction_op) { in ReductionOp()75 ncclRedOp_t reduction_op; in Run() local
68 std::unique_ptr<OpKernel> reduction_op( in BM_SegmentReduction() local
80 ncclRedOp_t reduction_op() const { return reduction_op_; } in reduction_op() function in tensorflow::__anon49c74ebe0111::NcclReduceOpBase
400 cub::Sum reduction_op; in CountAndSortParts() local
147 const ncclRedOp_t reduction_op; // applies when <type> is a reduction. member345 ncclRedOp_t reduction_op) { in AddToAllReduce()370 ncclRedOp_t reduction_op) { in AddReduceSend()376 ncclRedOp_t reduction_op) { in AddReduceRecv()398 ncclRedOp_t reduction_op) { in AddParticipant()
77 ncclRedOp_t reduction_op, TensorShape shape, in MakeReductionTestCase()263 ncclRedOp_t reduction_op = static_cast<ncclRedOp_t>(op); in TYPED_TEST() local426 ncclRedOp_t reduction_op = static_cast<ncclRedOp_t>(op); in TYPED_TEST() local
331 reduction_op='Add', argument