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 | 1566 auto sub_computation = module->AddEmbeddedComputation(sub_builder.Build()); in TEST_F() local
1572 HloInstruction::CreateCall(tuple_shape, {param}, sub_computation)); in TEST_F()
2433 HloComputation* sub_computation; in TEST_F() local
2442 sub_computation = module->AddEmbeddedComputation(builder.Build(add)); in TEST_F()
2450 HloInstruction::CreateCall(r0f32, {constant2}, sub_computation)); in TEST_F()
2452 HloInstruction::CreateCall(r0f32, {constant3}, sub_computation)); in TEST_F()
|
| D | dynamic_padder.cc | 1696 for (auto* sub_computation : hlo->called_computations()) { in DefaultAction() local
1697 for (auto* param : sub_computation->parameter_instructions()) { in DefaultAction()
|
| D | hlo_dataflow_analysis_test.cc | 2777 auto sub_computation = module_->AddEmbeddedComputation(sub_builder.Build()); in TEST_F() local
2778 sub_computation->CreateFusionInstruction({add, ones}, in TEST_F()
2789 HloInstruction::CreateCall(shape, {reverse}, sub_computation)); in TEST_F()
|