Home
last modified time | relevance | path

Searched defs:ToNcclReduction (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lhlo_gpu_to_tfrt_gpu/
H A Dccl_pattern.cc38 ncclRedOp_t ToNcclReduction(xla::ReductionKind kind) { in ToNcclReduction() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dnccl_utils.cc60 ncclRedOp_t ToNcclReduction(ReductionKind kind) { in ToNcclReduction() function