Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dnccl_all_reduce_thunk.cc167 CollectiveOpGroupMode GetGroupMode(OpT op) { in GetGroupMode() function
242 CollectiveOpGroupMode NcclAllReduceThunk::GetGroupMode( in GetGroupMode() function in xla::gpu::NcclAllReduceThunk
279 CollectiveOpGroupMode NcclAllReduceStartThunk::GetGroupMode( in GetGroupMode() function in xla::gpu::NcclAllReduceStartThunk
354 /*static*/ CollectiveOpGroupMode NcclReduceScatterThunk::GetGroupMode( in GetGroupMode() function in xla::gpu::NcclReduceScatterThunk
H A Dnccl_all_gather_thunk.h51 static CollectiveOpGroupMode GetGroupMode(mlir::lmhlo::AllGatherOp op) { in GetGroupMode() function
H A Dnccl_all_to_all_thunk.h51 static CollectiveOpGroupMode GetGroupMode(mlir::lmhlo::AllToAllOp op) { in GetGroupMode() function
H A Dnccl_collective_permute_thunk.h78 static CollectiveOpGroupMode GetGroupMode( in GetGroupMode() function