Searched refs:sub_computation (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | flatten_call_graph_test.cc | 229 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()
|
| D | buffer_assignment_test.cc | 1394 auto sub_computation = module->AddEmbeddedComputation(sub_builder.Build()); in TEST_F() local
1400 HloInstruction::CreateCall(tuple_shape, {param}, sub_computation)); in TEST_F()
2286 HloComputation* sub_computation; in TEST_F() local
2295 sub_computation = module->AddEmbeddedComputation(builder.Build(add)); in TEST_F()
2303 HloInstruction::CreateCall(r0f32, {constant2}, sub_computation)); in TEST_F()
2305 HloInstruction::CreateCall(r0f32, {constant3}, sub_computation)); in TEST_F()
|
| D | tuple_points_to_analysis_test.cc | 1238 auto sub_computation = module_->AddEmbeddedComputation(sub_builder.Build()); in TEST_F() local
1239 sub_computation->CreateFusionInstruction({add, ones}, in TEST_F()
1250 HloInstruction::CreateCall(shape, {reverse}, sub_computation)); in TEST_F()
|
| D | hlo_dataflow_analysis_test.cc | 2644 auto sub_computation = module_->AddEmbeddedComputation(sub_builder.Build()); in TEST_F() local
2645 sub_computation->CreateFusionInstruction({add, ones}, in TEST_F()
2656 HloInstruction::CreateCall(shape, {reverse}, sub_computation)); in TEST_F()
|