Searched refs:b_computation (Results 1 – 3 of 3) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | call_graph_test.cc | 321 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 D | flatten_call_graph_test.cc | 117 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 D | buffer_assignment_test.cc | 1696 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()
|