| /external/tensorflow/tensorflow/compiler/xla/service/ |
| 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()
|
| D | hlo_dce_test.cc | 225 auto called_computation = in TEST_F() local
233 HloInstruction::CreateCall(shape, {param}, called_computation)); in TEST_F()
337 auto called_computation = in TEST_F() local
345 ShapeUtil::MakeTokenShape(), {param}, called_computation)); in TEST_F()
|
| D | hlo_live_range.cc | 106 for (const HloComputation* called_computation : in FlattenSchedule() local
108 time = FlattenSchedule(*called_computation, time); in FlattenSchedule()
|
| D | memory_space_assignment.cc | 844 for (const HloComputation* called_computation : in FindAliases() local
847 called_computation->parameter_instructions()) { in FindAliases()
953 HloComputation* called_computation = in IsUseAllowedInAlternateMemory() local
957 called_computation->parameter_instruction(0); in IsUseAllowedInAlternateMemory()
964 if (parameter_use.instruction->parent() == called_computation && in IsUseAllowedInAlternateMemory()
977 << called_computation->name() in IsUseAllowedInAlternateMemory()
984 << called_computation->name() in IsUseAllowedInAlternateMemory()
1265 for (const HloComputation* called_computation : in AddRequiredAssignmentsForColocatedIntervals() local
1267 AddRequiredAssignment(called_computation->root_instruction(), in AddRequiredAssignmentsForColocatedIntervals()
1323 for (const HloComputation* called_computation : in AllocateAllocationValues() local
[all …]
|
| D | hlo_liveness_analysis.cc | 297 for (auto* called_computation : instruction->called_computations()) { in RunAnalysis() local
298 MarkLiveAtAllIndices(called_computation->root_instruction(), in RunAnalysis()
|
| D | call_graph_test.cc | 173 HloComputation* called_computation = in TEST_F() local
176 MakeCallingComputation(called_computation, /*callsites=*/3)); in TEST_F()
193 const CallGraphNode& called_node = call_graph->GetNode(called_computation); in TEST_F()
194 EXPECT_EQ(called_computation, called_node.computation()); in TEST_F()
|
| D | hlo_dataflow_analysis.cc | 57 for (const HloComputation* called_computation : in CalculatePostOrderScheduleHelper() local
59 ordinal = CalculatePostOrderScheduleHelper(called_computation, ordinal, in CalculatePostOrderScheduleHelper()
888 for (HloComputation* called_computation : user->called_computations()) { in Propagate()
890 call_graph_->GetNode(called_computation); in Propagate()
894 called_computation->parameter_instruction(operand_number)); in Propagate()
|
| D | hlo_alias_analysis_test.cc | 412 HloComputation* called_computation = in TEST_F() local
421 scalar_shape_, {constant1, constant2}, called_computation)); in TEST_F()
452 HloComputation* called_computation = in TEST_F() local
461 scalar_shape_, {constant1, constant2}, called_computation)); in TEST_F()
463 scalar_shape_, {call1, constant2}, called_computation)); in TEST_F()
|
| D | hlo_module.cc | 623 for (HloComputation* called_computation : in MakeComputationPostOrder()
625 nonroot_computations.insert(called_computation); in MakeComputationPostOrder()
|
| D | hlo_computation.cc | 370 for (HloComputation* called_computation : in ComputeComputationPostOrder()
372 ComputeComputationPostOrder(called_computation, visited, post_order); in ComputeComputationPostOrder()
|
| D | hlo_rematerialization.cc | 1946 for (HloComputation* called_computation : in RematerializeComputation()
1948 if (!ContainsKey(rematerialized_computations_, called_computation)) { in RematerializeComputation()
1955 RematerializeComputation(called_computation, schedule, in RematerializeComputation()
|
| D | hlo_dataflow_analysis_test.cc | 264 HloComputation* called_computation = in TEST_P() local
273 scalar_shape_, {constant1, constant2}, called_computation)); in TEST_P()
|