Home
last modified time | relevance | path

Searched refs:while_body (Results 1 – 25 of 47) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_invariant_code_motion_test.cc66 HloComputation* while_body = [&]() { in TEST_F() local
87 while_shape, MakeAlwaysTrueComputation(while_shape, m.get()), while_body, in TEST_F()
98 EXPECT_THAT(transformed_while->while_body()->instructions(), in TEST_F()
108 HloComputation* while_body = [&]() { in TEST_F() local
146 while_shape, MakeAlwaysTrueComputation(while_shape, m.get()), while_body, in TEST_F()
165 EXPECT_THAT(transformed_while->while_body()->instructions(), in TEST_F()
169 EXPECT_THAT(transformed_while->while_body()->instructions(), in TEST_F()
180 HloComputation* while_body = [&]() { in TEST_F() local
200 while_shape, MakeAlwaysTrueComputation(while_shape, m.get()), while_body, in TEST_F()
209 EXPECT_THAT(while_inst->while_body()->instructions(), Contains(op::Add())); in TEST_F()
[all …]
Dwhile_loop_simplifier_test.cc170 auto* true_op = while_op->while_body()->AddInstruction( in TEST_F()
173 while_op->while_body()->root_instruction())); in TEST_F()
187 auto* while_body = while_op->while_body(); in TEST_F() local
188 auto* token = while_body->AddInstruction(HloInstruction::CreateToken()); in TEST_F()
189 auto* send = while_body->AddInstruction(HloInstruction::CreateSend( in TEST_F()
190 while_body->AddInstruction( in TEST_F()
194 while_body->AddInstruction(HloInstruction::CreateSendDone(send)); in TEST_F()
203 auto* while_body = while_op->while_body(); in TEST_F() local
204 auto* token = while_body->AddInstruction(HloInstruction::CreateToken()); in TEST_F()
205 auto* recv = while_body->AddInstruction( in TEST_F()
[all …]
Dwhile_loop_simplifier.cc58 HloComputation* while_body = while_op->while_body(); in TryRemoveDeadWhileParams() local
59 HloInstruction* while_body_root = while_body->root_instruction(); in TryRemoveDeadWhileParams()
75 for (const HloInstruction* instr : {while_body->parameter_instruction(0), in TryRemoveDeadWhileParams()
97 for (HloComputation* comp : {while_body, while_cond}) { in TryRemoveDeadWhileParams()
135 operand->operand(0) != while_body->parameter_instruction(0) || in TryRemoveDeadWhileParams()
236 while_body_replacements = make_while_computation_replacements(while_body); in TryRemoveDeadWhileParams()
246 while_body->CloneWithReplacements(std::move(while_body_replacements)); in TryRemoveDeadWhileParams()
314 auto* while_body = while_op->while_body(); in TryRemoveConstantParams() local
316 auto* while_body_root = while_body->root_instruction(); in TryRemoveConstantParams()
323 TF_RET_CHECK(while_body->num_parameters() == 1); in TryRemoveConstantParams()
[all …]
Dwhile_loop_constant_sinking_test.cc64 auto* while_body = module->GetComputationWithName("body"); in TEST_F() local
65 EXPECT_THAT(while_body->root_instruction(), in TEST_F()
104 auto* while_body = module->GetComputationWithName("body"); in TEST_F() local
105 EXPECT_THAT(while_body->root_instruction(), in TEST_F()
145 auto* while_body = module->GetComputationWithName("body"); in TEST_F() local
146 EXPECT_THAT(while_body->root_instruction(), in TEST_F()
193 auto* while_body = module->GetComputationWithName("body"); in TEST_F() local
194 EXPECT_THAT(while_body->root_instruction(), in TEST_F()
235 auto* while_body = module->GetComputationWithName("body"); in TEST_F() local
236 EXPECT_THAT(while_body->root_instruction(), in TEST_F()
[all …]
Dwhile_loop_analysis.cc128 auto* while_body = while_op->while_body(); in GetLoopInductionVarTupleIdx() local
129 auto* while_body_root = while_body->root_instruction(); in GetLoopInductionVarTupleIdx()
137 auto* while_body_param = while_body->parameter_instruction(0); in GetLoopInductionVarTupleIdx()
250 auto* while_body = while_op->while_body(); in PatternMatchLoopTripCount() local
252 while_body->root_instruction()->operand(indvar_tuple_idx); in PatternMatchLoopTripCount()
364 auto* while_body = while_op->while_body(); in ComputeWhileLoopTripCount() local
366 while_body->root_instruction()->operand(*indvar_tuple_idx); in ComputeWhileLoopTripCount()
448 auto* while_body = while_op->while_body(); in ComputeWhileLoopTripCountUpperBound() local
449 auto* while_body_root = while_body->root_instruction(); in ComputeWhileLoopTripCountUpperBound()
Dwhile_loop_invariant_code_motion.cc43 HloComputation* while_body = while_instr->while_body(); in CreateLoopInvariantCopy() local
53 HloInstruction* while_body_param = while_body->parameter_instruction(0); in CreateLoopInvariantCopy()
154 HloComputation* while_body = while_instr->while_body(); in TryHoistingInvariantInstructionsFromWhileBody() local
170 for (auto* instr : WhileUtil::GetInvariantGTEsForWhileBody(*while_body)) { in TryHoistingInvariantInstructionsFromWhileBody()
192 for (auto* instruction : while_body->MakeInstructionPostOrder()) { in TryHoistingInvariantInstructionsFromWhileBody()
203 for (auto* instruction : while_body->MakeInstructionPostOrder()) { in TryHoistingInvariantInstructionsFromWhileBody()
284 live_in_instructions_result.new_while_instr->while_body(); in TryHoistingInvariantInstructionsFromWhileBody()
Dwhile_loop_constant_sinking.cc52 HloComputation* while_body = while_instr->while_body(); in TrySinkingConstantsIntoWhileLoop() local
65 WhileUtil::GetInvariantGTEsForWhileBody(*while_body); in TrySinkingConstantsIntoWhileLoop()
80 while_body->AddInstruction(invariant_value.Clone(/*suffix=*/".sunk")); in TrySinkingConstantsIntoWhileLoop()
82 invariant_body_gte, constant_instr, while_body->root_instruction(), in TrySinkingConstantsIntoWhileLoop()
Dwhile_util_test.cc90 EXPECT_THAT(new_while_instr->while_body()->root_instruction(), in TEST()
124 EXPECT_THAT(new_while_instr->while_body()->root_instruction(), in TEST()
157 HloComputation* while_body = module->GetComputationWithName("body"); in TEST() local
159 ASSERT_NE(while_body, nullptr) in TEST()
163 WhileUtil::GetInvariantGTEsForWhileBody(*while_body); in TEST()
Dhlo_ordering.cc61 const HloComputation* body = a_ancestor->while_body(); in ExecutesBefore()
148 a.defining_instruction(), b.defining_instruction()->while_body()) || in IsDefinedBefore()
197 xla_while->while_body()) || in UseIsBeforeValueDefinition()
212 xla_while->while_body()) || in UseIsBeforeValueDefinition()
Dwhile_util.cc116 WidenWhileBody(while_instr->while_body(), new_while_shape)); in MakeInstructionsLiveIn()
260 const HloComputation& while_body) { in GetInvariantGTEsForWhileBody() argument
263 while_body.root_instruction()->operands(); in GetInvariantGTEsForWhileBody()
268 instr->operand(0) == while_body.parameter_instruction(0)) { in GetInvariantGTEsForWhileBody()
Dbfloat16_propagation.cc172 HloComputation* body = while_hlo->while_body(); in DetermineWhileComputationsPrecision()
262 use.instruction->while_body()->parameter_instruction( in AllUsersConsumeBF16()
331 caller_counts_[hlo->while_body()] > 1)) { in DetermineInstructionPrecision()
444 adjust_computation(hlo->while_body(), hlo->operands()); in AdjustCalledComputationParameters()
494 adjust_computation(hlo->while_body(), hlo); in AdjustCalledComputationRoot()
566 ResolveInconsistencyOfAliasingBuffersHelper(hlo->while_body(), in ResolveInconsistencyOfAliasingBuffersHelper()
Dcopy_insertion_test.cc843 while_hlo->while_body()->root_instruction(), in TEST_F()
1060 EXPECT_EQ(CountCopies(*while_hlo->while_body()), 0); in TEST_F()
1077 EXPECT_EQ(CountCopies(*while_hlo->while_body()), 0); in TEST_F()
1124 EXPECT_EQ(CountCopies(*while_hlo->while_body()), 1); in TEST_F()
1125 if (while_hlo->while_body() in TEST_F()
1131 while_hlo->while_body()->root_instruction(), in TEST_F()
1135 while_hlo->while_body()->root_instruction(), in TEST_F()
1188 EXPECT_EQ(CountCopies(*while_hlo->while_body()), 1); in TEST_F()
1189 if (while_hlo->while_body() in TEST_F()
1195 while_hlo->while_body()->root_instruction(), in TEST_F()
[all …]
Dwhile_util.h89 const HloComputation& while_body);
Dhlo_verifier.cc648 CheckParameterCount(xla_while, xla_while->while_body(), 1)); in HandleWhile()
652 CheckOperandAndParameter(xla_while, 0, xla_while->while_body(), 0)); in HandleWhile()
666 xla_while->while_body()->root_instruction()->shape()); in HandleWhile()
1355 auto* while_body = xla_while->while_body(); in HandleWhile() local
1361 if (while_body->num_parameters() != 1) { in HandleWhile()
1364 while_body->num_parameters(), while_body->ToString()); in HandleWhile()
Dhlo_module_dce.cc48 auto* while_body_comp = xla_while->while_body(); in RunWhileDCE()
Dhlo_liveness_analysis.cc178 MarkLiveAtIndex(instruction->while_body()->root_instruction(), shape_index, in PropagateLivenessThroughWhile()
212 MarkLiveAtIndex(xla_while->while_body()->root_instruction(), in PropagateLivenessToParameterCallers()
Dar_crs_combiner_test.cc247 auto body_tuple = root_while->while_body()->root_instruction(); in TEST_F()
285 auto body_tuple = root_while->while_body()->root_instruction(); in TEST_F()
323 auto body_tuple = root_while->while_body()->root_instruction(); in TEST_F()
Dflatten_call_graph.cc41 CHECK_EQ(computation, instruction->while_body()); in ReplaceCalledComputation()
Dhlo_dataflow_analysis.cc501 callsite.instruction()->while_body()->root_instruction()) { in UpdateParameterValueSet()
503 callsite.instruction()->while_body()->root_instruction())); in UpdateParameterValueSet()
589 &GetInstructionValueSet(xla_while->while_body()->root_instruction()), in UpdateWhileValueSet()
715 callsite.instruction()->while_body()->parameter_instruction(0)); in Propagate()
Dhlo_module_dce_test.cc53 auto* while_body_comp = instruction->while_body(); in WhileBodyHasPassThroughTupleElement()
/external/swiftshader/third_party/llvm-7.0/llvm/test/Transforms/CodeGenPrepare/AArch64/
Dlarge-offset-gep.ll12 %phi = phi i32 [ 0, %entry ], [ %i, %while_body ]
18 br i1 %cmp, label %while_body, label %while_end
20 while_body:
38 %phi = phi i32 [ 0, %entry ], [ %i, %while_body ]
44 br i1 %cmp1, label %while_body, label %while_end
46 while_body:
71 %phi = phi i32 [ 0, %if_end ], [ %i, %while_body ]
77 br i1 %cmp1, label %while_body, label %while_end
79 while_body:
98 %phi = phi i32 [ 0, %entry ], [ %i, %while_body ]
[all …]
/external/tensorflow/tensorflow/python/autograph/operators/
Dcontrol_flow.py98 def while_body(iterate_index, *state): function
115 while_body,
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dcontrol_flow_ops.py64 def while_body(i, *ta_list): function
88 while_body,
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dfor_thunk.cc54 profiler->FinishHloComputation(hlo_instruction()->while_body()); in ExecuteOnStream()
Dwhile_thunk.cc88 profiler->FinishHloComputation(hlo_instruction()->while_body()); in ExecuteOnStream()

12