Searched defs:reduction_op (Results 1 – 13 of 13) sorted by relevance
45 Status ReductionOp(const string& merge_op, ncclRedOp_t* reduction_op) { in ReductionOp()140 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
71 std::unique_ptr<OpKernel> reduction_op( in BM_UnsortedSegmentReduction() local135 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::__anon05def67a0111::NcclReduceOpBase
405 gpuprim::Sum reduction_op; in CountAndSortParts() local
102 ncclRedOp_t reduction_op = ToNcclReduction(*reduction_kind); in CclOpConversionRewrite() local
405 nir_op reduction_op; in vtn_handle_subgroup() local
119 mlir::Operation* reduction_op = result_op->getOperand(0).getDefiningOp(); in FindReductionOp() local176 StatusOr<mlir::Operation*> reduction_op = FindReductionOp(block); in MatchAllReduceComputation() local
79 auto reduction_op = llvm::dyn_cast_or_null<mlir::TF::DTensorAllReduceOp>( in CheckReduceAndSumOptimizationCriteria() local
498 def test_unsorted_segment_reduction(self, reduction_op): argument
880 nir_op reduction_op = nir_intrinsic_reduction_op(instr); in print_intrinsic_instr() local
3189 auto reduction_op = in FoldOperandsPermutation() local