| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | root_instruction_sinker_test.cc | 128 auto called_computation = in TEST_F() local
130 int num_instructions = called_computation->instruction_count(); in TEST_F()
160 auto called_computation = in TEST_F() local
162 const auto& sequence = module->schedule().sequence(called_computation); in TEST_F()
164 called_computation->root_instruction()); in TEST_F()
165 EXPECT_THAT(called_computation->root_instruction(), in TEST_F()
|
| H A D | hlo_instructions.cc | 88 void SetThreadName(HloComputation* called_computation, in SetThreadName() argument
91 called_computation->SetExecutionThread(execution_thread); in SetThreadName()
92 for (HloInstruction* instr : called_computation->instructions()) { in SetThreadName()
1561 HloComputation* called_computation, absl::string_view prefix) in HloCallableInstruction() argument
1567 AppendComputation(called_computation); in HloCallableInstruction()
1579 for (auto called_computation : called_computations) { in HloCallableInstruction() local
1580 AppendComputation(called_computation); in HloCallableInstruction()
1586 HloComputation* HloCallableInstruction::called_computation() const { in called_computation() function in xla::HloCallableInstruction
1592 return called_computation()->root_instruction(); in called_computation_root()
1598 called_computation()->parameter_instructions().size()); in AddCallOperand()
[all …]
|
| H A D | call_graph_test.cc | 176 HloComputation* called_computation = in TEST_F() local
179 MakeCallingComputation(called_computation, /*callsites=*/3)); in TEST_F()
197 const CallGraphNode& called_node = call_graph->GetNode(called_computation); in TEST_F()
198 EXPECT_EQ(called_computation, called_node.computation()); in TEST_F()
203 EXPECT_EQ(3, call_graph->GetComputationCallers(called_computation).size()); in TEST_F()
|
| H A D | hlo_live_range.cc | 98 for (const HloComputation* called_computation : in FlattenSchedule() local
102 FlattenSchedule(*called_computation, in FlattenSchedule()
104 ? called_computation in FlattenSchedule()
|
| H A D | hlo_dce_test.cc | 223 auto called_computation = in TEST_F() local
231 HloInstruction::CreateCall(shape, {param}, called_computation)); in TEST_F()
335 auto called_computation = in TEST_F() local
343 ShapeUtil::MakeTokenShape(), {param}, called_computation)); in TEST_F()
|
| H A D | memory_space_assignment.cc | 1127 for (const HloComputation* called_computation : in FindAliases() local
1130 called_computation->parameter_instructions()) { in FindAliases()
1236 HloComputation* called_computation = in IsUseAllowedInAlternateMemory() local
1240 called_computation->parameter_instruction(0); in IsUseAllowedInAlternateMemory()
1247 if (parameter_use.instruction->parent() == called_computation && in IsUseAllowedInAlternateMemory()
1260 << called_computation->name() in IsUseAllowedInAlternateMemory()
1267 << called_computation->name() in IsUseAllowedInAlternateMemory()
1609 for (const HloComputation* called_computation : in AddRequiredAssignmentsForColocatedIntervals() local
1611 AddRequiredAssignment(called_computation->root_instruction(), in AddRequiredAssignmentsForColocatedIntervals()
1709 for (const HloComputation* called_computation : in AllocateAllocationValues() local
[all …]
|
| H A D | hlo_alias_analysis_test.cc | 444 HloComputation* called_computation = in TEST_F() local
453 scalar_shape_, {constant1, constant2}, called_computation)); in TEST_F()
484 HloComputation* called_computation = in TEST_F() local
493 scalar_shape_, {constant1, constant2}, called_computation)); in TEST_F()
495 scalar_shape_, {call1, constant2}, called_computation)); in TEST_F()
|
| H A D | hlo_liveness_analysis.cc | 316 for (auto* called_computation : instruction->called_computations()) { in RunAnalysis() local
317 MarkLiveAtAllIndices(called_computation->root_instruction(), in RunAnalysis()
|
| H A D | hlo_dataflow_analysis.cc | 56 for (const HloComputation* called_computation : in CalculatePostOrderScheduleHelper() local
58 ordinal = CalculatePostOrderScheduleHelper(called_computation, ordinal, in CalculatePostOrderScheduleHelper()
1277 for (HloComputation* called_computation : user->called_computations()) { in Propagate()
1279 call_graph_->GetNode(called_computation); in Propagate()
1283 called_computation->parameter_instruction(operand_number)); in Propagate()
|
| H A D | conditional_simplifier.cc | 597 for (const HloComputation* called_computation : in InstructionCallsChannelInstructions() local
599 if (ComputationCallsChannelInstructions(*called_computation)) { in InstructionCallsChannelInstructions()
|
| H A D | hlo_dataflow_analysis_test.cc | 271 HloComputation* called_computation = in TEST_P() local
280 scalar_shape_, {constant1, constant2}, called_computation)); in TEST_P()
1166 const HloComputation* called_computation = in TEST_P() local
1169 called_computation->parameter_instruction(0); in TEST_P()
1174 called_computation->parameter_instruction(1); in TEST_P()
1178 const HloInstruction* root = called_computation->root_instruction(); in TEST_P()
|
| H A D | hlo_module.cc | 743 for (HloComputation* called_computation : in MakeComputationPostOrder()
745 nonroot_computations.insert(called_computation); in MakeComputationPostOrder()
|
| H A D | hlo_rematerialization.cc | 2026 for (HloComputation* called_computation : in RematerializeComputation()
2028 if (!ContainsKey(rematerialized_computations_, called_computation)) { in RematerializeComputation()
2035 RematerializeComputation(called_computation, schedule, in RematerializeComputation()
|
| H A D | hlo_instructions.h | 1151 HloComputation* called_computation,
1184 HloComputation* called_computation() const;
1344 HloComputation* called_computation);
|
| H A D | hlo_computation.cc | 417 for (HloComputation* called_computation : in ComputeComputationPostOrder()
419 ComputeComputationPostOrder(called_computation, visited, post_order); in ComputeComputationPostOrder()
|
| H A D | conditional_code_motion.cc | 663 for (const HloComputation* called_computation : in MoveInstructionOut() local
665 HloInstruction* root = called_computation->root_instruction(); in MoveInstructionOut()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_compiler.cc | 1099 HloComputation* called_computation = *it; in SubcomputationEmissionOrder() local
1101 called_computation, c.allow_reassociation || allow_reassociation}; in SubcomputationEmissionOrder()
|