Home
last modified time | relevance | path

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

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