Home
last modified time | relevance | path

Searched refs:called_computation (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_dce_test.cc187 auto called_computation = in TEST_F() local
195 HloInstruction::CreateCall(shape, {param}, called_computation)); in TEST_F()
297 auto called_computation = in TEST_F() local
305 HloInstruction::CreateCall(shape, {param}, called_computation)); in TEST_F()
Dcall_graph_test.cc173 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()
Dhlo_liveness_analysis.cc297 for (auto* called_computation : instruction->called_computations()) { in RunAnalysis() local
298 MarkLiveAtAllIndices(called_computation->root_instruction(), in RunAnalysis()
Dhlo_alias_analysis_test.cc420 HloComputation* called_computation = in TEST_F() local
429 scalar_shape_, {constant1, constant2}, called_computation)); in TEST_F()
460 HloComputation* called_computation = in TEST_F() local
469 scalar_shape_, {constant1, constant2}, called_computation)); in TEST_F()
471 scalar_shape_, {call1, constant2}, called_computation)); in TEST_F()
Dheap_simulator.cc299 for (const HloComputation* called_computation : in RunComputation() local
302 schedule_->sequence(called_computation); in RunComputation()
304 *called_computation, called_sequence, points_to_analysis)); in RunComputation()
Dhlo_dataflow_analysis.cc689 for (HloComputation* called_computation : user->called_computations()) { in Propagate()
691 call_graph_->GetNode(called_computation); in Propagate()
695 called_computation->parameter_instruction(operand_number)); in Propagate()
Dhlo_rematerialization.cc1145 for (HloComputation* called_computation : in RematerializeComputation()
1147 if (!ContainsKey(rematerialized_computations_, called_computation)) { in RematerializeComputation()
1154 RematerializeComputation(called_computation, schedule, in RematerializeComputation()
Dhlo_module.cc506 for (HloComputation* called_computation : in MakeComputationPostOrder()
508 nonroot_computations.insert(called_computation); in MakeComputationPostOrder()
Dhlo_dataflow_analysis_test.cc261 HloComputation* called_computation = in TEST_P() local
270 scalar_shape_, {constant1, constant2}, called_computation)); in TEST_P()
311 HloComputation* called_computation = in TEST_P() local
320 scalar_shape_, {constant1, constant2}, called_computation)); in TEST_P()
322 scalar_shape_, {constant1, constant2}, called_computation)); in TEST_P()
366 HloComputation* called_computation = in TEST_P() local
375 scalar_shape_, {constant1, constant2}, called_computation)); in TEST_P()
377 scalar_shape_, {call1, constant2}, called_computation)); in TEST_P()
Dhlo_computation.cc305 for (HloComputation* called_computation : in ComputeComputationPostOrder()
307 ComputeComputationPostOrder(called_computation, visited, post_order); in ComputeComputationPostOrder()