Home
last modified time | relevance | path

Searched refs:c_computation (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dcall_graph_test.cc319 HloComputation* c_computation = in TEST_F() local
322 MakeMappingComputation(c_computation, /*callsites=*/1)); in TEST_F()
330 HloInstruction::CreateCall(kScalarShape, {param0}, c_computation)); in TEST_F()
353 const CallGraphNode& c_node = call_graph->GetNode(c_computation); in TEST_F()
398 EXPECT_LT(index_of(c_computation), index_of(b_computation)); in TEST_F()
407 EXPECT_TRUE(call_graph->Dominates(entry_computation, c_computation)); in TEST_F()
411 EXPECT_FALSE(call_graph->Dominates(c_computation, entry_computation)); in TEST_F()
417 EXPECT_TRUE(call_graph->Dominates(a_computation, c_computation)); in TEST_F()
419 EXPECT_FALSE(call_graph->Dominates(c_computation, a_computation)); in TEST_F()
423 EXPECT_FALSE(call_graph->Dominates(b_computation, c_computation)); in TEST_F()
[all …]
H A Dflatten_call_graph_test.cc115 HloComputation* c_computation = in TEST_F() local
118 MakeMappingComputation(c_computation, /*callsites=*/1)); in TEST_F()
126 HloInstruction::CreateCall(kScalarShape, {param0}, c_computation)); in TEST_F()
146 const CallGraphNode& c_node = flat_call_graph->GetNode(c_computation); in TEST_F()
206 HloComputation* c_computation = in TEST_F() local
210 MakeCallingComputation(c_computation, /*callsites=*/2, ".B")); in TEST_F()
220 const CallGraphNode& c_node = call_graph->GetNode(c_computation); in TEST_F()
H A Dbuffer_assignment_test.cc1689 auto c_computation = c_builder.Build(); in TEST_F() local
1695 HloInstruction::CreateCall(tuple_shape, {b_param}, c_computation.get())); in TEST_F()
1710 module->AddEmbeddedComputation(std::move(c_computation)); in TEST_F()