Home
last modified time | relevance | path

Searched refs:sub_computation (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dflatten_call_graph_test.cc229 HloComputation* sub_computation = in TEST_F() local
242 kScalarShape, pred, constant1, sub_computation, constant2, in TEST_F()
243 sub_computation)); in TEST_F()
253 const CallGraphNode& sub_node = call_graph->GetNode(sub_computation); in TEST_F()
H A Dbuffer_assignment_test.cc1643 auto sub_computation = module->AddEmbeddedComputation(sub_builder.Build()); in TEST_F() local
1649 HloInstruction::CreateCall(tuple_shape, {param}, sub_computation)); in TEST_F()
2511 HloComputation* sub_computation; in TEST_F() local
2520 sub_computation = module->AddEmbeddedComputation(builder.Build(add)); in TEST_F()
2528 HloInstruction::CreateCall(r0f32, {constant2}, sub_computation)); in TEST_F()
2530 HloInstruction::CreateCall(r0f32, {constant3}, sub_computation)); in TEST_F()
H A Ddynamic_padder.cc2042 for (auto* sub_computation : hlo->called_computations()) { in DefaultAction() local
2043 for (auto* param : sub_computation->parameter_instructions()) { in DefaultAction()
H A Dhlo_dataflow_analysis_test.cc2842 auto sub_computation = module->AddEmbeddedComputation(sub_builder.Build()); in TEST_F() local
2843 sub_computation->CreateFusionInstruction({add, ones}, in TEST_F()
2854 HloInstruction::CreateCall(shape, {reverse}, sub_computation)); in TEST_F()