Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dcall_graph_test.cc316 HloComputation* b_computation = module->AddEmbeddedComputation( in TEST_F() local
327 kScalarShape, cond_computation, b_computation, call)); in TEST_F()
347 const CallGraphNode& b_node = call_graph->GetNode(b_computation); in TEST_F()
369 UnorderedElementsAre(a_computation, b_computation)); in TEST_F()
394 EXPECT_LT(index_of(c_computation), index_of(b_computation)); in TEST_F()
395 EXPECT_LT(index_of(b_computation), index_of(a_computation)); in TEST_F()
402 EXPECT_TRUE(call_graph->Dominates(entry_computation, b_computation)); in TEST_F()
406 EXPECT_FALSE(call_graph->Dominates(b_computation, entry_computation)); in TEST_F()
412 EXPECT_TRUE(call_graph->Dominates(a_computation, b_computation)); in TEST_F()
414 EXPECT_FALSE(call_graph->Dominates(b_computation, a_computation)); in TEST_F()
[all …]
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()
Dbuffer_assignment_test.cc1619 auto b_computation = b_builder.Build(); in TEST_F() local
1627 HloInstruction::CreateCall(tuple_shape, {a_tuple}, b_computation.get())); in TEST_F()
1635 module->AddEmbeddedComputation(std::move(b_computation)); in TEST_F()