Searched refs:c_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 | 319 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 D | flatten_call_graph_test.cc | 115 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 D | buffer_assignment_test.cc | 1689 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()
|