Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dcall_graph_test.cc321 HloComputation* b_computation = module->AddEmbeddedComputation( in TEST_F() local
332 kScalarShape, cond_computation, b_computation, call)); in TEST_F()
352 const CallGraphNode& b_node = call_graph->GetNode(b_computation); in TEST_F()
373 UnorderedElementsAre(a_computation, b_computation)); in TEST_F()
398 EXPECT_LT(index_of(c_computation), index_of(b_computation)); in TEST_F()
399 EXPECT_LT(index_of(b_computation), index_of(a_computation)); in TEST_F()
406 EXPECT_TRUE(call_graph->Dominates(entry_computation, b_computation)); in TEST_F()
410 EXPECT_FALSE(call_graph->Dominates(b_computation, entry_computation)); in TEST_F()
416 EXPECT_TRUE(call_graph->Dominates(a_computation, b_computation)); in TEST_F()
418 EXPECT_FALSE(call_graph->Dominates(b_computation, a_computation)); in TEST_F()
[all …]
H A Dflatten_call_graph_test.cc117 HloComputation* b_computation = module->AddEmbeddedComputation( in TEST_F() local
128 kScalarShape, cond_computation, b_computation, call)); in TEST_F()
209 HloComputation* b_computation = module->AddEmbeddedComputation( in TEST_F() local
213 MakeCallingComputation(b_computation, /*callsites=*/2, ".Entry")); in TEST_F()
223 const CallGraphNode& b_node = call_graph->GetNode(b_computation); in TEST_F()
H A Dbuffer_assignment_test.cc1696 auto b_computation = b_builder.Build(); in TEST_F() local
1704 HloInstruction::CreateCall(tuple_shape, {a_tuple}, b_computation.get())); in TEST_F()
1712 module->AddEmbeddedComputation(std::move(b_computation)); in TEST_F()