Searched refs:AllReduceCombiner (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | all_reduce_combiner_test.cc | 125 AllReduceCombiner combine(10 * 1024 * 1024, kMaxCombineCount); in TEST_F()
173 AllReduceCombiner combine(10 * 1024 * 1024, kMaxCombineCount); in TEST_F()
195 AllReduceCombiner combine((8 + 4) * 1024 - 1, kMaxCombineCount); in TEST_F()
205 AllReduceCombiner combine((8 + 4) * 1024, kMaxCombineCount); in TEST_F()
232 AllReduceCombiner combine(1024 * 1024, kMaxCombineCount); in TEST_F()
261 AllReduceCombiner combine(1024 * 1024, kMaxCombineCount); in TEST_F()
297 AllReduceCombiner combine(1024 * 1024, kMaxCombineCount); in TEST_F()
342 AllReduceCombiner combine(1024 * 1024, kMaxCombineCount); in TEST_F()
372 AllReduceCombiner combine(1024 * 1024, kMaxCombineCount); in TEST_F()
412 AllReduceCombiner combine(1024 * 1024, kMaxCombineCount); in TEST_F()
[all …]
|
| D | all_reduce_combiner.h | 32 class AllReduceCombiner : public HloModulePass {
34 AllReduceCombiner(int64 combine_threshold_in_bytes,
|
| D | all_reduce_combiner.cc | 262 AllReduceCombiner::AllReduceCombiner(int64 combine_threshold_in_bytes, in AllReduceCombiner() function in xla::AllReduceCombiner
267 StatusOr<bool> AllReduceCombiner::Run(HloModule* module) { in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_compiler.cc | 340 pipeline.AddPass<AllReduceCombiner>( in OptimizeHloModule()
|