Searched defs:CanImplement (Results 1 – 5 of 5) sorted by relevance
141 bool CanImplement(OpT op) { in CanImplement() function232 bool NcclAllReduceThunk::CanImplement(mlir::lmhlo::AllReduceOp op) { in CanImplement() function in xla::gpu::NcclAllReduceThunk268 bool NcclAllReduceStartThunk::CanImplement( in CanImplement() function in xla::gpu::NcclAllReduceStartThunk343 /*static*/ bool NcclReduceScatterThunk::CanImplement( in CanImplement() function in xla::gpu::NcclReduceScatterThunk
47 /*static*/ bool NcclAllGatherThunk::CanImplement(mlir::lmhlo::AllGatherOp op) { in CanImplement() function in xla::gpu::NcclAllGatherThunk
51 /*static*/ bool NcclAllToAllThunk::CanImplement(mlir::lmhlo::AllToAllOp op) { in CanImplement() function in xla::gpu::NcclAllToAllThunk
99 /*static*/ bool NcclCollectivePermuteThunk::CanImplement( in CanImplement() function in xla::gpu::NcclCollectivePermuteThunk
1195 static bool CanImplement(AllGatherOp op) { in CanImplement() function in tensorflow::CollectiveOpLowering1198 static bool CanImplement(AllReduceOp op) { in CanImplement() function in tensorflow::CollectiveOpLowering1201 static bool CanImplement(AllReduceStartOp op) { in CanImplement() function in tensorflow::CollectiveOpLowering1204 static bool CanImplement(ReduceScatterOp op) { in CanImplement() function in tensorflow::CollectiveOpLowering1207 static bool CanImplement(AllToAllOp op) { in CanImplement() function in tensorflow::CollectiveOpLowering1210 static bool CanImplement(CollectivePermuteOp op) { in CanImplement() function in tensorflow::CollectiveOpLowering