Home
last modified time | relevance | path

Searched defs:CanImplement (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_all_reduce_thunk.cc141 bool CanImplement(OpT op) { in CanImplement() function
232 bool NcclAllReduceThunk::CanImplement(mlir::lmhlo::AllReduceOp op) { in CanImplement() function in xla::gpu::NcclAllReduceThunk
264 bool NcclAllReduceStartThunk::CanImplement( in CanImplement() function in xla::gpu::NcclAllReduceStartThunk
335 /*static*/ bool NcclReduceScatterThunk::CanImplement( in CanImplement() function in xla::gpu::NcclReduceScatterThunk
Dnccl_all_gather_thunk.cc43 /*static*/ bool NcclAllGatherThunk::CanImplement(mlir::lmhlo::AllGatherOp op) { in CanImplement() function in xla::gpu::NcclAllGatherThunk
Dnccl_all_to_all_thunk.cc47 /*static*/ bool NcclAllToAllThunk::CanImplement(mlir::lmhlo::AllToAllOp op) { in CanImplement() function in xla::gpu::NcclAllToAllThunk
Dnccl_collective_permute_thunk.cc92 /*static*/ bool NcclCollectivePermuteThunk::CanImplement( in CanImplement() function in xla::gpu::NcclCollectivePermuteThunk