Home
last modified time | relevance | path

Searched refs:AllGatherDecomposer (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dall_gather_decomposer.h27 class AllGatherDecomposer : public HloModulePass {
29 explicit AllGatherDecomposer( in AllGatherDecomposer() function
32 AllGatherDecomposer() in AllGatherDecomposer() function
Dall_gather_decomposer_test.cc51 AllGatherDecomposer decomposer; in TEST_F()
74 AllGatherDecomposer decomposer; in TEST_F()
97 AllGatherDecomposer decomposer; in TEST_F()
120 AllGatherDecomposer decomposer; in TEST_F()
146 AllGatherDecomposer decomposer; in TEST_F()
Dall_gather_decomposer.cc140 StatusOr<bool> AllGatherDecomposer::Run(HloModule* module) { in Run()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc309 pipeline.AddPass<AllGatherDecomposer>(); in RunHloPassesThroughLayoutAssn()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_compiler.cc157 pipeline.AddPass<AllGatherDecomposer>( in OptimizeHloModule()