Home
last modified time | relevance | path

Searched defs:called_computation (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Droot_instruction_sinker_test.cc128 auto called_computation = in TEST_F() local
160 auto called_computation = in TEST_F() local
H A Dhlo_dce_test.cc223 auto called_computation = in TEST_F() local
335 auto called_computation = in TEST_F() local
H A Dmemory_space_assignment.cc1127 for (const HloComputation* called_computation : in FindAliases() local
1236 HloComputation* called_computation = in IsUseAllowedInAlternateMemory() local
1609 for (const HloComputation* called_computation : in AddRequiredAssignmentsForColocatedIntervals() local
1709 for (const HloComputation* called_computation : in AllocateAllocationValues() local
1751 for (const HloComputation* called_computation : in AllocateAllocationValues() local
4274 for (const HloComputation* called_computation : in VerifyAndExportHeapSimulatorTrace() local
H A Dhlo_alias_analysis_test.cc444 HloComputation* called_computation = in TEST_F() local
484 HloComputation* called_computation = in TEST_F() local
H A Dhlo_live_range.cc98 for (const HloComputation* called_computation : in FlattenSchedule() local
H A Dhlo_liveness_analysis.cc316 for (auto* called_computation : instruction->called_computations()) { in RunAnalysis() local
H A Dhlo_instructions.cc88 void SetThreadName(HloComputation* called_computation, in SetThreadName()
1561 HloComputation* called_computation, absl::string_view prefix) in HloCallableInstruction()
1579 for (auto called_computation : called_computations) { in HloCallableInstruction() local
1586 HloComputation* HloCallableInstruction::called_computation() const { in called_computation() function in xla::HloCallableInstruction
2155 HloComputation* called_computation) in HloCallInstruction()
H A Dconditional_simplifier.cc597 for (const HloComputation* called_computation : in InstructionCallsChannelInstructions() local
H A Dcall_graph_test.cc176 HloComputation* called_computation = in TEST_F() local
H A Dhlo_dataflow_analysis_test.cc271 HloComputation* called_computation = in TEST_P() local
1166 const HloComputation* called_computation = in TEST_P() local
H A Dconditional_code_motion.cc663 for (const HloComputation* called_computation : in MoveInstructionOut() local
H A Dhlo_dataflow_analysis.cc56 for (const HloComputation* called_computation : in CalculatePostOrderScheduleHelper() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_compiler.cc1099 HloComputation* called_computation = *it; in SubcomputationEmissionOrder() local