Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_concat_code_motion.h60 class WhileLoopConcatCodeMotion : public HloModulePass {
62 explicit WhileLoopConcatCodeMotion(int64_t min_operand_count_to_optimize) in WhileLoopConcatCodeMotion() function
64 ~WhileLoopConcatCodeMotion() override = default;
Dwhile_loop_concat_code_motion_test.cc82 WhileLoopConcatCodeMotion(2).Run(module.get())); in TEST_F()
138 WhileLoopConcatCodeMotion(2).Run(module.get())); in TEST_F()
189 WhileLoopConcatCodeMotion(2).Run(module.get())); in TEST_F()
257 WhileLoopConcatCodeMotion(2).Run(module.get())); in TEST_F()
329 WhileLoopConcatCodeMotion(2).Run(module.get())); in TEST_F()
419 WhileLoopConcatCodeMotion(2).Run(module.get())); in TEST_F()
Dwhile_loop_concat_code_motion.cc1017 StatusOr<bool> WhileLoopConcatCodeMotion::Run(HloModule* module) { in Run()