Searched refs:WhileUtil (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_util_test.cc | 64 TEST(WhileUtil, MakeZeroInstructionsLiveOp) { in TEST() argument
76 WhileUtil::MakeInstructionsLiveInResult make_live_in_result, in TEST()
77 WhileUtil::MakeInstructionsLiveIn(while_instr, /*instructions=*/{})); in TEST()
107 WhileUtil::MakeInstructionsLiveInResult make_live_in_result, in TEST()
108 WhileUtil::MakeInstructionsLiveIn(while_instr, in TEST()
163 WhileUtil::GetInvariantGTEsForWhileBody(*while_body); in TEST()
203 WhileUtil::MakeInstructionsLiveInResult make_live_in_result, in TEST()
204 WhileUtil::MakeInstructionsLiveIn(while_instr, in TEST()
|
| D | while_util.cc | 96 /*static*/ StatusOr<WhileUtil::MakeInstructionsLiveInResult>
97 WhileUtil::MakeInstructionsLiveIn( in MakeInstructionsLiveIn()
142 WhileUtil::MakeInstructionsLiveInResult result; in MakeInstructionsLiveIn()
176 const std::function<StatusOr<WhileUtil::LoopStateTy>( in MakeCountedLoopBodyComputation()
177 HloInstruction*, const WhileUtil::LoopStateTy&)>& loop_body_generator) { in MakeCountedLoopBodyComputation()
206 HloComputation* computation, const WhileUtil::LoopStateTy& init_values) { in MakeInitTupleFromInitValues()
217 static Shape MakeLoopStateShape(const WhileUtil::LoopStateTy& init_values) { in MakeLoopStateShape()
227 /*static*/ StatusOr<WhileUtil::LoopStateTy> WhileUtil::MakeCountedLoop( in MakeCountedLoop()
229 const WhileUtil::LoopStateTy& init_values, in MakeCountedLoop()
230 const WhileUtil::LoopBodyGeneratorTy& loop_body_generator, in MakeCountedLoop()
[all …]
|
| D | while_loop_invariant_code_motion.cc | 170 for (auto* instr : WhileUtil::GetInvariantGTEsForWhileBody(*while_body)) { in TryHoistingInvariantInstructionsFromWhileBody()
280 WhileUtil::MakeInstructionsLiveInResult live_in_instructions_result, in TryHoistingInvariantInstructionsFromWhileBody()
281 WhileUtil::MakeInstructionsLiveIn(while_instr, replacement_instructions)); in TryHoistingInvariantInstructionsFromWhileBody()
|
| D | while_loop_constant_sinking.cc | 63 WhileUtil::GetGTEsMapForWhileConditional(*while_cond); in TrySinkingConstantsIntoWhileLoop()
65 WhileUtil::GetInvariantGTEsForWhileBody(*while_body); in TrySinkingConstantsIntoWhileLoop()
|
| D | while_util.h | 25 class WhileUtil {
|
| D | dynamic_dimension_inference.cc | 490 WhileUtil::MakeInstructionsLiveInResult result, in HandleWhile()
491 WhileUtil::MakeInstructionsLiveIn(hlo, operands_to_add)); in HandleWhile()
|
| D | gather_expander.cc | 348 WhileUtil::MakeCountedLoop( in ExpandInstruction()
|
| D | scatter_expander.cc | 398 WhileUtil::MakeCountedLoop( in ExpandScatter()
|