/external/tensorflow/tensorflow/compiler/xla/service/ |
D | while_loop_constant_sinking_test.cc | 281 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()
|
D | hlo_ordering.cc | 62 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()
|
D | while_loop_simplifier_test.cc | 429 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()
|
D | while_loop_analysis.cc | 110 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()
|
D | while_loop_simplifier.cc | 57 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()
|
D | bfloat16_propagation.cc | 174 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()
|
D | hlo_liveness_analysis.cc | 186 MarkLiveAtIndex(instruction->while_condition()->root_instruction(), {}, in PropagateLivenessThroughWhile() 238 MarkLiveAtIndex(caller->while_condition()->root_instruction(), {}, in PropagateLivenessThroughControlFlow()
|
D | flatten_call_graph.cc | 38 if (computation == instruction->while_condition()) { in ReplaceCalledComputation()
|
D | while_loop_constant_sinking.cc | 51 HloComputation* while_cond = while_instr->while_condition(); in TrySinkingConstantsIntoWhileLoop()
|
D | call_inliner_test.cc | 113 computation->root_instruction()->while_condition()->root_instruction(), in TEST_F()
|
D | hlo_verifier.cc | 650 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()
|
D | hlo_schedule_test.cc | 255 HloComputation* cond = xla_while->while_condition(); in TEST_F()
|
D | while_util.cc | 110 WidenWhileCondition(while_instr->while_condition(), new_while_shape)); in MakeInstructionsLiveIn()
|
D | hlo_instruction.cc | 1475 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()
|
D | hlo_instruction.h | 1064 HloComputation* while_condition() const; 1066 void set_while_condition(HloComputation* while_condition);
|
D | hlo_module_group_metadata.cc | 302 tracked_instructions_[hlo->while_condition()] = in RecordInstructions()
|
D | dynamic_dimension_inference.cc | 523 hlo->while_condition(), binding_for_while, parent_)); in HandleWhile()
|
D | hlo_module.cc | 149 replacements, instruction->while_condition(), nullptr); in ReplaceComputations()
|
D | hlo_cost_analysis.cc | 672 ProcessUnnestedSubcomputation(xla_while->while_condition())); in HandleWhile()
|
D | layout_assignment.cc | 552 HloComputation* condition = instruction->while_condition(); in AddMandatoryConstraints() 937 FindOrDie(computation_layouts_, instruction->while_condition()), in CheckLayouts()
|
D | hlo_dataflow_analysis.cc | 717 callsite.instruction()->while_condition()->parameter_instruction( in Propagate()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | while_thunk.cc | 63 profiler->FinishHloComputation(hlo_instruction()->while_condition()); in ExecuteOnStream()
|
D | ir_emitter_unnested.cc | 978 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/ |
D | ir_emitter.cc | 2298 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()
|
D | cpu_compiler.cc | 232 xla_while->while_condition()->Accept(&candidates_for_condition)); in HandleWhile()
|