Home
last modified time | relevance | path

Searched refs:kEliminateAllGathers (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dgather_expander_test.cc46 Status status = GatherExpander{GatherExpander::kEliminateAllGathers} in TEST_F()
76 GatherExpander{GatherExpander::kEliminateAllGathers}.Run(module.get())); in TEST_F()
139 GatherExpander{GatherExpander::kEliminateAllGathers}.Run(module.get())); in TEST_F()
199 GatherExpander pass(GatherExpander::kEliminateAllGathers); in TEST_F()
Dgather_expander.h42 kEliminateAllGathers, enumerator
Dgather_expander.cc389 (mode_ == kEliminateAllGathers || GatherLoopTripCount(inst) == 1); in InstructionMatchesPattern()