Home
last modified time | relevance | path

Searched refs:while_condition (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_constant_sinking_test.cc281 auto* while_condition = module->GetComputationWithName("condition"); in TEST_F() local
282 EXPECT_THAT(while_condition->root_instruction(), op::Lt(_, op::Constant())); in TEST_F()
321 auto* while_condition = module->GetComputationWithName("condition"); in TEST_F() local
322 EXPECT_THAT(while_condition->root_instruction(), in TEST_F()
364 auto* while_condition = module->GetComputationWithName("condition"); in TEST_F() local
365 EXPECT_THAT(while_condition->root_instruction(), op::Lt(_, op::Constant())); in TEST_F()
366 for (const HloInstruction* inst : while_condition->instructions()) { in TEST_F()
414 auto* while_condition = module->GetComputationWithName("condition"); in TEST_F() local
415 EXPECT_THAT(while_condition->root_instruction(), in TEST_F()
Dhlo_ordering.cc62 const HloComputation* condition = a_ancestor->while_condition(); in ExecutesBefore()
151 b.defining_instruction()->while_condition()))) { in IsDefinedBefore()
199 xla_while->while_condition())) { in UseIsBeforeValueDefinition()
214 xla_while->while_condition())) { in UseIsBeforeValueDefinition()
Dwhile_loop_simplifier_test.cc429 EXPECT_THAT(new_while_op->while_condition()->root_instruction(), in TEST_F()
565 new_while->while_condition()->parameter_instruction(0)->shape(), in TEST_F()
650 new_while->while_condition()->parameter_instruction(0)->shape(), in TEST_F()
720 new_while->while_condition()->parameter_instruction(0)->shape(), in TEST_F()
728 EXPECT_THAT(new_while->while_condition()->root_instruction(), in TEST_F()
Dwhile_loop_analysis.cc110 auto* while_cond = while_op->while_condition(); in GetLoopInductionVarTupleIdx()
264 auto* while_cond = while_op->while_condition(); in PatternMatchLoopTripCount()
369 auto* while_cond = while_op->while_condition(); in ComputeWhileLoopTripCount()
438 auto* while_cond = while_op->while_condition(); in ComputeWhileLoopTripCountUpperBound()
Dwhile_loop_simplifier.cc57 HloComputation* while_cond = while_op->while_condition(); in TryRemoveDeadWhileParams()
315 auto* while_cond = while_op->while_condition(); in TryRemoveConstantParams()
570 propagate_constant(while_op->while_condition())); in TryPropagateConstant()
647 auto* while_cond = while_op->while_condition(); in TryFlattenNestedTuples()
774 auto* while_cond = while_op->while_condition(); in TryMergeInductionVariables()
1003 ContainsInstrWithOpcode(while_op->while_condition(), in Run()
1027 ContainsInstrWithOpcode(while_op->while_condition(), in Run()
Dbfloat16_propagation.cc174 HloComputation* condition = while_hlo->while_condition(); in DetermineWhileComputationsPrecision()
256 use.instruction->while_condition()->parameter_instruction( in AllUsersConsumeBF16()
330 (caller_counts_[hlo->while_condition()] > 1 || in DetermineInstructionPrecision()
443 adjust_computation(hlo->while_condition(), hlo->operands()); in AdjustCalledComputationParameters()
564 while (ResolveInconsistencyOfAliasingBuffersHelper(hlo->while_condition(), in ResolveInconsistencyOfAliasingBuffersHelper()
Dhlo_liveness_analysis.cc186 MarkLiveAtIndex(instruction->while_condition()->root_instruction(), {}, in PropagateLivenessThroughWhile()
238 MarkLiveAtIndex(caller->while_condition()->root_instruction(), {}, in PropagateLivenessThroughControlFlow()
Dflatten_call_graph.cc38 if (computation == instruction->while_condition()) { in ReplaceCalledComputation()
Dwhile_loop_constant_sinking.cc51 HloComputation* while_cond = while_instr->while_condition(); in TrySinkingConstantsIntoWhileLoop()
Dcall_inliner_test.cc113 computation->root_instruction()->while_condition()->root_instruction(), in TEST_F()
Dhlo_verifier.cc650 CheckParameterCount(xla_while, xla_while->while_condition(), 1)); in HandleWhile()
654 CheckOperandAndParameter(xla_while, 0, xla_while->while_condition(), 0)); in HandleWhile()
656 xla_while->while_condition()->root_instruction()->shape(); in HandleWhile()
1354 auto* while_cond = xla_while->while_condition(); in HandleWhile()
Dhlo_schedule_test.cc255 HloComputation* cond = xla_while->while_condition(); in TEST_F()
Dwhile_util.cc110 WidenWhileCondition(while_instr->while_condition(), new_while_shape)); in MakeInstructionsLiveIn()
Dhlo_instruction.cc1475 CreateWhile(shape, while_condition(), while_body(), new_operands[0]); in CloneWithNewOperands()
1795 eq_computations(while_condition(), other.while_condition())); in IdenticalSlowPath()
2026 HloComputation* HloInstruction::while_condition() const { in while_condition() function in xla::HloInstruction
2283 StrCat("condition=", PrintName(while_condition()->name(), options))); in ExtraAttributesToString()
2329 StrCat("condition=\n", while_condition()->ToString(new_options))); in ExtraAttributesToString()
Dhlo_instruction.h1064 HloComputation* while_condition() const;
1066 void set_while_condition(HloComputation* while_condition);
Dhlo_module_group_metadata.cc302 tracked_instructions_[hlo->while_condition()] = in RecordInstructions()
Ddynamic_dimension_inference.cc523 hlo->while_condition(), binding_for_while, parent_)); in HandleWhile()
Dhlo_module.cc149 replacements, instruction->while_condition(), nullptr); in ReplaceComputations()
Dhlo_cost_analysis.cc672 ProcessUnnestedSubcomputation(xla_while->while_condition())); in HandleWhile()
Dlayout_assignment.cc552 HloComputation* condition = instruction->while_condition(); in AddMandatoryConstraints()
937 FindOrDie(computation_layouts_, instruction->while_condition()), in CheckLayouts()
Dhlo_dataflow_analysis.cc717 callsite.instruction()->while_condition()->parameter_instruction( in Propagate()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dwhile_thunk.cc63 profiler->FinishHloComputation(hlo_instruction()->while_condition()); in ExecuteOnStream()
Dir_emitter_unnested.cc978 HloComputation* condition = xla_while->while_condition(); in HandleWhile()
2061 xla_while->while_condition()->parameter_instruction(0); in CheckWhileBuffersShareAllocation()
2118 HloComputation* condition = hlo->while_condition(); in BuildWhileThunk()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2298 HloComputation* condition = xla_while->while_condition(); in HandleWhile()
2325 xla_while, xla_while->while_condition()->parameter_instruction(0), in HandleWhile()
2355 EmitGlobalCall(*xla_while->while_condition(), IrName(xla_while, "cond")); in HandleWhile()
2357 Load(GetBufferForGlobalCallReturnValue(*xla_while->while_condition())), in HandleWhile()
Dcpu_compiler.cc232 xla_while->while_condition()->Accept(&candidates_for_condition)); in HandleWhile()

12