Home
last modified time | relevance | path

Searched refs:CanImplement (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_all_gather_thunk.h44 static bool CanImplement(const HloInstruction* hlo);
45 static bool CanImplement(mlir::lmhlo::AllGatherOp op);
Dnccl_all_gather_thunk.cc40 /*static*/ bool NcclAllGatherThunk::CanImplement(const HloInstruction* hlo) { in CanImplement() function in xla::gpu::NcclAllGatherThunk
51 /*static*/ bool NcclAllGatherThunk::CanImplement(mlir::lmhlo::AllGatherOp op) { in CanImplement() function in xla::gpu::NcclAllGatherThunk
Dnccl_all_reduce_thunk.h43 static bool CanImplement(mlir::lmhlo::AllReduceOp op);
Dnccl_all_to_all_thunk.h43 static bool CanImplement(mlir::lmhlo::AllToAllOp op);
Dnccl_all_to_all_thunk.cc43 /*static*/ bool NcclAllToAllThunk::CanImplement(mlir::lmhlo::AllToAllOp op) { in CanImplement() function in xla::gpu::NcclAllToAllThunk
Dnccl_all_reduce_thunk.cc100 /*static*/ bool NcclAllReduceThunk::CanImplement(mlir::lmhlo::AllReduceOp op) { in CanImplement() function in xla::gpu::NcclAllReduceThunk
Dgpu_compiler.cc159 return !NcclAllGatherThunk::CanImplement(&ag); in OptimizeHloModule()
Dir_emitter_unnested.cc3112 replica_count > 1 && NcclThunkType::CanImplement(op); in EmitNcclThunkFromMlir()