Home
last modified time | relevance | path

Searched refs:combine_threshold_in_bytes (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dall_reduce_combiner.cc109 AllReduceCombiner::AllReduceCombiner(int64_t combine_threshold_in_bytes, in AllReduceCombiner() argument
111 : combine_threshold_in_bytes_(combine_threshold_in_bytes), in AllReduceCombiner()
H A Dreduce_scatter_combiner.cc112 ReduceScatterCombiner::ReduceScatterCombiner(int64_t combine_threshold_in_bytes, in ReduceScatterCombiner() argument
114 : combine_threshold_in_bytes_(combine_threshold_in_bytes), in ReduceScatterCombiner()
H A Dall_gather_combiner.cc135 AllGatherCombiner::AllGatherCombiner(int64_t combine_threshold_in_bytes, in AllGatherCombiner() argument
137 : combine_threshold_in_bytes_(combine_threshold_in_bytes), in AllGatherCombiner()
H A Dall_gather_combiner.h34 AllGatherCombiner(int64_t combine_threshold_in_bytes,
H A Dall_reduce_combiner.h34 AllReduceCombiner(int64_t combine_threshold_in_bytes,
H A Dreduce_scatter_combiner.h31 ReduceScatterCombiner(int64_t combine_threshold_in_bytes,