Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dconditional_code_motion_test.cc75 ConditionalCodeMotion pass(true, true); in TEST_F()
124 ConditionalCodeMotion pass(true, true); in TEST_F()
160 ConditionalCodeMotion pass(true, true); in TEST_F()
201 ConditionalCodeMotion pass(true, true); in TEST_F()
242 ConditionalCodeMotion pass(true, true); in TEST_F()
278 ConditionalCodeMotion pass(true, true); in TEST_F()
347 ConditionalCodeMotion pass(true, true); in TEST_F()
416 ConditionalCodeMotion pass(true, true); in TEST_F()
484 ConditionalCodeMotion pass(true, true); in TEST_F()
576 ConditionalCodeMotion pass(true, true); in TEST_F()
[all …]
H A Dconditional_code_motion.h87 class ConditionalCodeMotion : public HloModulePass {
102 explicit ConditionalCodeMotion(bool is_layout_sensitive,
115 explicit ConditionalCodeMotion(bool is_layout_sensitive, in ConditionalCodeMotion() function
H A Dconditional_code_motion.cc616 StatusOr<bool> ConditionalCodeMotion::MoveInstructionOut( in MoveInstructionOut()
723 StatusOr<bool> ConditionalCodeMotion::MoveInstructionIn( in MoveInstructionIn()
932 int c = ConditionalCodeMotion::flip_start(search_config_); in FlipMutation()
940 auto flip_count = ConditionalCodeMotion::DecrementMaxFlip(&search_config_); in FlipMutation()
956 auto flip_stride = ConditionalCodeMotion::flip_stride(search_config_); in FlipMutation()
1357 ConditionalCodeMotion::Decision ConditionalCodeMotion::ConsiderCodeMotion( in ConsiderCodeMotion()
1392 StatusOr<bool> ConditionalCodeMotion::Run( in Run()
1651 void ConditionalCodeMotion::SetDefaultMoveConfig() { in SetDefaultMoveConfig()
1713 void ConditionalCodeMotion::ParseSearchConfiguration( in ParseSearchConfiguration()