Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dcall_graph_test.cc319 HloComputation* a_computation; in TEST_F() local
328 a_computation = module->AddEmbeddedComputation(builder.Build()); in TEST_F()
337 kScalarShape, cond_computation, a_computation, param0)); in TEST_F()
346 const CallGraphNode& a_node = call_graph->GetNode(a_computation); in TEST_F()
364 UnorderedElementsAre(cond_computation, a_computation)); in TEST_F()
369 UnorderedElementsAre(a_computation, b_computation)); in TEST_F()
393 EXPECT_LT(index_of(cond_computation), index_of(a_computation)); in TEST_F()
395 EXPECT_LT(index_of(b_computation), index_of(a_computation)); in TEST_F()
401 EXPECT_TRUE(call_graph->Dominates(entry_computation, a_computation)); in TEST_F()
405 EXPECT_FALSE(call_graph->Dominates(a_computation, entry_computation)); in TEST_F()
[all …]
Dflatten_call_graph_test.cc120 HloComputation* a_computation; in TEST_F() local
129 a_computation = module->AddEmbeddedComputation(builder.Build()); in TEST_F()
138 kScalarShape, cond_computation, a_computation, param0)); in TEST_F()
Dbuffer_assignment_test.cc1628 auto a_computation = a_builder.Build(); in TEST_F() local
1634 module->AddEntryComputation(std::move(a_computation)); in TEST_F()