Searched defs:IsDegenerate (Results 1 – 6 of 6) sorted by relevance
161 bool IsDegenerate(OpT op, int64_t replica_count, int64_t partition_count) { in IsDegenerate() function236 bool NcclAllReduceThunk::IsDegenerate(mlir::lmhlo::AllReduceOp op, in IsDegenerate() function in xla::gpu::NcclAllReduceThunk269 bool NcclAllReduceStartThunk::IsDegenerate(mlir::lmhlo_gpu::AllReduceStartOp op, in IsDegenerate() function in xla::gpu::NcclAllReduceStartThunk340 /*static*/ bool NcclReduceScatterThunk::IsDegenerate( in IsDegenerate() function in xla::gpu::NcclReduceScatterThunk
46 static bool IsDegenerate(mlir::lmhlo::AllGatherOp op, int64_t replica_count, in IsDegenerate() function
46 static bool IsDegenerate(mlir::lmhlo::AllToAllOp op, int64_t replica_count, in IsDegenerate() function
74 bool NcclCollectiveConfig::IsDegenerate(int64_t replica_count, in IsDegenerate() function in xla::gpu::NcclCollectiveConfig
75 /*static*/ bool NcclCollectivePermuteThunk::IsDegenerate( in IsDegenerate() function in xla::gpu::NcclCollectivePermuteThunk
227 bool IsDegenerate = false) in GetProcessAttributesFunc()