/external/tensorflow/tensorflow/compiler/xla/service/ |
D | while_loop_invariant_code_motion_test.cc | 66 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 …]
|
D | while_loop_simplifier_test.cc | 170 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 …]
|
D | while_loop_simplifier.cc | 58 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 …]
|
D | while_loop_constant_sinking_test.cc | 64 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 …]
|
D | while_loop_analysis.cc | 128 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()
|
D | while_loop_invariant_code_motion.cc | 43 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()
|
D | while_loop_constant_sinking.cc | 52 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()
|
D | while_util_test.cc | 90 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()
|
D | hlo_ordering.cc | 61 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()
|
D | while_util.cc | 116 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()
|
D | bfloat16_propagation.cc | 172 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()
|
D | copy_insertion_test.cc | 843 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 …]
|
D | while_util.h | 89 const HloComputation& while_body);
|
D | hlo_verifier.cc | 648 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()
|
D | hlo_module_dce.cc | 48 auto* while_body_comp = xla_while->while_body(); in RunWhileDCE()
|
D | hlo_liveness_analysis.cc | 178 MarkLiveAtIndex(instruction->while_body()->root_instruction(), shape_index, in PropagateLivenessThroughWhile() 212 MarkLiveAtIndex(xla_while->while_body()->root_instruction(), in PropagateLivenessToParameterCallers()
|
D | ar_crs_combiner_test.cc | 247 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()
|
D | flatten_call_graph.cc | 41 CHECK_EQ(computation, instruction->while_body()); in ReplaceCalledComputation()
|
D | hlo_dataflow_analysis.cc | 501 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()
|
D | hlo_module_dce_test.cc | 53 auto* while_body_comp = instruction->while_body(); in WhileBodyHasPassThroughTupleElement()
|
/external/swiftshader/third_party/llvm-7.0/llvm/test/Transforms/CodeGenPrepare/AArch64/ |
D | large-offset-gep.ll | 12 %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/ |
D | control_flow.py | 98 def while_body(iterate_index, *state): function 115 while_body,
|
/external/tensorflow/tensorflow/python/ops/parallel_for/ |
D | control_flow_ops.py | 64 def while_body(i, *ta_list): function 88 while_body,
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | for_thunk.cc | 54 profiler->FinishHloComputation(hlo_instruction()->while_body()); in ExecuteOnStream()
|
D | while_thunk.cc | 88 profiler->FinishHloComputation(hlo_instruction()->while_body()); in ExecuteOnStream()
|