Searched defs:CanImplement (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ | ||
| D | nccl_all_gather_thunk.cc | 40 /*static*/ bool NcclAllGatherThunk::CanImplement(const HloInstruction* hlo) { in CanImplement() function in xla::gpu::NcclAllGatherThunk |
| D | nccl_all_to_all_thunk.cc | 43 /*static*/ bool NcclAllToAllThunk::CanImplement(mlir::lmhlo::AllToAllOp op) { in CanImplement() function in xla::gpu::NcclAllToAllThunk |
| D | nccl_all_reduce_thunk.cc | 100 /*static*/ bool NcclAllReduceThunk::CanImplement(mlir::lmhlo::AllReduceOp op) { in CanImplement() function in xla::gpu::NcclAllReduceThunk |