Home
last modified time | relevance | path

Searched defs:reduction_kind (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_all_reduce_thunk.h33 ReductionKind reduction_kind; member
Dnccl_all_reduce_thunk.cc149 absl::optional<ReductionKind> reduction_kind = in GetNcclAllReduceConfig() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dreduce_scatter_combiner.cc74 absl::optional<ReductionKind> reduction_kind = in CombineReduceScatters() local
Dcollective_ops_utils.h258 ReductionKind reduction_kind; member
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lhlo_gpu_to_tfrt_gpu/
Dccl_pattern.cc94 auto reduction_kind = in CclOpConversionRewrite() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_runtime.cc507 xla::ReductionKind reduction_kind = participant.reduction_kind; in DoAllReduce() local
568 T GetInitialValue(xla::ReductionKind reduction_kind) { in GetInitialValue()
582 T PerformReductionStep(xla::ReductionKind reduction_kind, T a, T b) { in PerformReductionStep()
684 xla::int32 channel_id_present, xla::int64 op_id, xla::int32 reduction_kind, in __xla_cpu_runtime_AllReduce()