Searched defs:reduction_op (Results 1 – 11 of 11) sorted by relevance
41 Status ReductionOp(const string& merge_op, ncclRedOp_t* reduction_op) { in ReductionOp()137 ncclRedOp_t reduction_op; in Enqueue() local
191 const ncclRedOp_t reduction_op; // applies when <type> is a reduction. member434 ncclRedOp_t reduction_op) { in AddToAllReduce()459 ncclRedOp_t reduction_op) { in AddReduceSend()465 ncclRedOp_t reduction_op) { in AddReduceRecv()492 ncclRedOp_t reduction_op) { in AddParticipant()
105 ncclRedOp_t reduction_op, TensorShape shape, in MakeReductionTestCase()323 ncclRedOp_t reduction_op = static_cast<ncclRedOp_t>(op); in RunMultiNodeAllReduceTest() local471 ncclRedOp_t reduction_op = static_cast<ncclRedOp_t>(op); in TYPED_TEST() local875 ncclRedOp_t reduction_op = static_cast<ncclRedOp_t>(0); in TYPED_TEST() local
45 mlir::Operation* reduction_op = &block.front(); in MatchReductionComputation() local
405 nir_op reduction_op; in vtn_handle_subgroup() local
67 std::unique_ptr<OpKernel> reduction_op( in BM_SegmentReduction() local
84 ncclRedOp_t reduction_op() const { return reduction_op_; } in reduction_op() function in tensorflow::__anon27d3229c0111::NcclReduceOpBase
416 gpuprim::Sum reduction_op; in CountAndSortParts() local
491 def test_unsorted_segment_reduction(self, reduction_op): argument
880 nir_op reduction_op = nir_intrinsic_reduction_op(instr); in print_intrinsic_instr() local
2777 auto reduction_op = in FoldOperandsPermutation() local