/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_module_test.cc | 73 module->AddEmbeddedComputation(CreateConstantComputation()); in TEST_F() 88 module->AddEmbeddedComputation(CreateConstantComputation()); in TEST_F() 90 module->AddEmbeddedComputation(CreateCallComputation({computation1})); in TEST_F() 92 module->AddEmbeddedComputation(CreateCallComputation({computation1})); in TEST_F() 118 fused_computation = module->AddEmbeddedComputation(b.Build()); in TEST_F() 154 module->AddEmbeddedComputation(CreateConstantComputation()); in TEST_F() 156 module->AddEmbeddedComputation(CreateCallComputation({computation1})); in TEST_F() 158 module->AddEmbeddedComputation(CreateCallComputation({computation1})); in TEST_F()
|
D | flatten_call_graph_test.cc | 113 module->AddEmbeddedComputation(MakeConditionComputation()); in TEST_F() 115 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F() 116 HloComputation* b_computation = module->AddEmbeddedComputation( in TEST_F() 128 a_computation = module->AddEmbeddedComputation(builder.Build()); in TEST_F() 164 cond_computation = module->AddEmbeddedComputation(builder.Build()); in TEST_F() 206 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F() 208 HloComputation* b_computation = module->AddEmbeddedComputation( in TEST_F() 229 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F()
|
D | call_graph_test.cc | 119 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F() 139 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F() 168 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F() 201 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F() 244 module->AddEmbeddedComputation(MakeScalarComputation(HloOpcode::kCeil)); in TEST_F() 246 module->AddEmbeddedComputation(MakeScalarComputation(HloOpcode::kFloor)); in TEST_F() 303 module->AddEmbeddedComputation(MakeConditionComputation()); in TEST_F() 305 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F() 306 HloComputation* b_computation = module->AddEmbeddedComputation( in TEST_F() 318 a_computation = module->AddEmbeddedComputation(builder.Build()); in TEST_F() [all …]
|
D | copy_insertion_test.cc | 749 auto condition = module_->AddEmbeddedComputation( in BuildWhileInstructionWithCustomInit() 751 auto body = module_->AddEmbeddedComputation( in BuildWhileInstructionWithCustomInit() 783 auto condition = module_->AddEmbeddedComputation( in TEST_F() 786 module_->AddEmbeddedComputation(BuildIndependentBodyComputation()); in TEST_F() 812 auto condition = module_->AddEmbeddedComputation( in TEST_F() 814 auto body = module_->AddEmbeddedComputation(BuildDependentBodyComputation()); in TEST_F() 856 auto condition = module_->AddEmbeddedComputation( in TEST_F() 858 auto body = module_->AddEmbeddedComputation( in TEST_F() 872 auto condition1 = module_->AddEmbeddedComputation( in TEST_F() 874 auto condition2 = module_->AddEmbeddedComputation( in TEST_F() [all …]
|
D | hlo_subcomputation_unification_test.cc | 73 module->AddEmbeddedComputation(CreateR0S32IdentityComputation()); in TEST_F() 75 module->AddEmbeddedComputation(CreateR0S32IdentityComputation()); in TEST_F() 110 module->AddEmbeddedComputation(CreateR0S32AdditionComputation()); in TEST_F() 112 module->AddEmbeddedComputation(CreateR0S32AdditionComputation()); in TEST_F() 150 module->AddEmbeddedComputation(CreateR1S32AdditionComputation(r1s32_5_)); in TEST_F() 152 module->AddEmbeddedComputation(CreateR1S32AdditionComputation(r1s32_3_)); in TEST_F() 197 module->AddEmbeddedComputation(builder.Build()); in TEST_F()
|
D | while_loop_simplifier_test.cc | 63 condition = module->AddEmbeddedComputation(cond_builder.Build()); in MakeSimpleLoop() 88 body = module->AddEmbeddedComputation(body_builder.Build()); in MakeSimpleLoop() 104 return module->AddEmbeddedComputation(builder.Build()); in MakeAlwaysTrueComputation() 196 condition = module().AddEmbeddedComputation(cond_builder.Build()); in TEST_F() 208 body = module().AddEmbeddedComputation(body_builder.Build()); in TEST_F() 244 body = module().AddEmbeddedComputation(body_builder.Build()); in TEST_F() 274 body = module().AddEmbeddedComputation(body_builder.Build()); in TEST_F() 297 body = module().AddEmbeddedComputation(body_builder.Build()); in TEST_F() 330 body = module().AddEmbeddedComputation(body_builder.Build()); in TEST_F() 366 condition = module().AddEmbeddedComputation(cond_builder.Build()); in TEST_F() [all …]
|
D | call_inliner_test.cc | 57 module->AddEmbeddedComputation(inner.Build()); in TEST_F() 92 module->AddEmbeddedComputation(just_false.Build()); in TEST_F() 98 module->AddEmbeddedComputation(call_false_builder.Build()); in TEST_F() 131 module->AddEmbeddedComputation(just_false.Build()); in TEST_F() 154 auto outfeed_computation = module->AddEmbeddedComputation(outfeeder.Build()); in TEST_F()
|
D | buffer_assignment_test.cc | 583 module->AddEmbeddedComputation(BuildMapComputationPlus1("f32+1")); in TEST_F() 638 module->AddEmbeddedComputation(BuildMapComputationPlus1("f32+1")); in TEST_F() 689 module->AddEmbeddedComputation(BuildWhileConditionComputation("if<4")); in TEST_F() 691 module->AddEmbeddedComputation(BuildWhileBodyComputation("add-update")); in TEST_F() 754 auto true_computation = module->AddEmbeddedComputation( in TEST_F() 756 auto false_computation = module->AddEmbeddedComputation( in TEST_F() 1045 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F() 1053 auto call_computation = module->AddEmbeddedComputation(call_builder.Build()); in TEST_F() 1229 auto sub_computation = module->AddEmbeddedComputation(sub_builder.Build()); in TEST_F() 1293 module->AddEmbeddedComputation(std::move(d_computation)); in TEST_F() [all …]
|
D | while_loop_invariant_code_motion_test.cc | 56 return module->AddEmbeddedComputation(builder.Build()); in MakeAlwaysTrueComputation() 78 return module().AddEmbeddedComputation(builder.Build()); in TEST_F() 137 return module().AddEmbeddedComputation(builder.Build()); in TEST_F() 191 return module().AddEmbeddedComputation(builder.Build()); in TEST_F() 230 return module().AddEmbeddedComputation(builder.Build()); in TEST_F() 264 return module().AddEmbeddedComputation(builder.Build()); in TEST_F() 305 return module().AddEmbeddedComputation(builder.Build()); in TEST_F() 350 return module().AddEmbeddedComputation(builder.Build()); in TEST_F() 399 while_body = module().AddEmbeddedComputation(builder.Build()); in TEST_F() 422 HloComputation* result = module().AddEmbeddedComputation(builder.Build()); in TEST_F()
|
D | hlo_alias_analysis_test.cc | 227 module_->AddEmbeddedComputation(subbuilder.Build()); in TEST_F() 266 module_->AddEmbeddedComputation(subbuilder.Build()); in TEST_F() 342 HloComputation* body = module_->AddEmbeddedComputation(body_builder.Build()); in TEST_F() 351 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_F() 436 HloComputation* body = module_->AddEmbeddedComputation(body_builder.Build()); in TEST_F() 444 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_F() 508 module_->AddEmbeddedComputation(build_cond_computation()); in TEST_F() 510 module_->AddEmbeddedComputation(build_cond_computation()); in TEST_F() 525 module_->AddEmbeddedComputation(inner_builder.Build()); in TEST_F() 542 module_->AddEmbeddedComputation(outer_builder.Build()); in TEST_F() [all …]
|
D | hlo_ordering_test.cc | 109 module->AddEmbeddedComputation(builder_c.Build()); in TEST_F() 117 module->AddEmbeddedComputation(builder_b.Build()); in TEST_F() 123 module->AddEmbeddedComputation(builder_a.Build()); in TEST_F() 183 HloComputation* body = module->AddEmbeddedComputation(body_builder.Build()); in TEST_F() 191 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F() 246 HloComputation* body = module->AddEmbeddedComputation(body_builder.Build()); in TEST_F() 254 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
|
D | hlo_rematerialization_test.cc | 220 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F() 222 HloComputation* body_computation = module->AddEmbeddedComputation( in TEST_F() 259 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F() 261 HloComputation* body_computation = module->AddEmbeddedComputation( in TEST_F() 294 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F() 296 HloComputation* inner_computation = module->AddEmbeddedComputation( in TEST_F() 299 module->AddEmbeddedComputation(MakeRematerializableWhileComputation( in TEST_F() 431 subcomputation = module->AddEmbeddedComputation(builder.Build()); in TEST_F() 547 subcomputation = module->AddEmbeddedComputation(builder.Build()); in TEST_P()
|
D | hlo_dataflow_analysis_test.cc | 258 module_->AddEmbeddedComputation(subbuilder.Build()); in TEST_P() 307 module_->AddEmbeddedComputation(subbuilder.Build()); in TEST_P() 361 module_->AddEmbeddedComputation(subbuilder.Build()); in TEST_P() 414 module_->AddEmbeddedComputation(inner_builder.Build()); in TEST_P() 425 module_->AddEmbeddedComputation(outer_builder.Build()); in TEST_P() 489 HloComputation* body = module_->AddEmbeddedComputation(body_builder.Build()); in TEST_P() 498 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_P() 591 HloComputation* body = module_->AddEmbeddedComputation(body_builder.Build()); in TEST_P() 599 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_P() 658 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_P() [all …]
|
D | hlo_dce_test.cc | 189 module->AddEmbeddedComputation(callee_builder.Build()); in TEST_F() 230 auto cond_computation = module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F() 243 auto body_computation = module->AddEmbeddedComputation(body_builder.Build()); in TEST_F() 285 module->AddEmbeddedComputation(nested_callee_builder.Build()); in TEST_F() 296 module->AddEmbeddedComputation(callee_builder.Build()); in TEST_F() 337 auto reduce_subcomp = module->AddEmbeddedComputation(subcomp_builder.Build()); in TEST_F() 378 auto reduce_subcomp = module->AddEmbeddedComputation(subcomp_builder.Build()); in TEST_F()
|
D | graphviz_example.cc | 53 return module->AddEmbeddedComputation(builder.Build()); in AddScalarConstantComputation() 66 return module->AddEmbeddedComputation(builder.Build()); in ScalarSumComputation() 78 return module->AddEmbeddedComputation(builder.Build()); in CallForwardingComputation()
|
D | inliner_test.cc | 63 hlo_module->AddEmbeddedComputation(std::move(max_f32)); in TEST_F() 97 hlo_module->AddEmbeddedComputation(std::move(const2_f32)); in TEST_F() 135 hlo_module->AddEmbeddedComputation(std::move(max_f32)); in TEST_F()
|
D | hlo_scheduling_test.cc | 54 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F() 61 module->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
|
D | flatten_call_graph.cc | 87 module->AddEmbeddedComputation(computation->Clone()); in FlattenNode() 102 module->AddEmbeddedComputation(callee->Clone()); in FlattenNode()
|
D | while_util.cc | 39 return narrow_condition->parent()->AddEmbeddedComputation(builder.Build()); in WidenWhileCondition() 64 return narrow_body->parent()->AddEmbeddedComputation(builder.Build()); in WidenWhileBody()
|
D | hlo_module.cc | 108 HloComputation* HloModule::AddEmbeddedComputation( in AddEmbeddedComputation() function in xla::HloModule 458 HloComputation* nested_computation = AddEmbeddedComputation( in OutlineExpressionFromComputation() 559 module->AddEmbeddedComputation(std::move(cloned_computation)); in Clone() 582 HloComputation* clone = AddEmbeddedComputation(computation->Clone("", this)); in DeepCloneComputation()
|
D | buffer_liveness_test.cc | 403 module->AddEmbeddedComputation(embedded_builder.Build()); in TEST_F() 515 module->AddEmbeddedComputation(builder.Build()); in TEST_F() 577 module->AddEmbeddedComputation(builder.Build()); in TEST_F() 650 auto* computation = module->AddEmbeddedComputation(builder.Build()); in Run() 778 module->AddEmbeddedComputation(builder.Build()); in Run()
|
D | hlo_computation_test.cc | 78 module->AddEmbeddedComputation(CreateNegateComputation()); in TEST_F() 90 module->AddEmbeddedComputation(CreateNegateComputation()); in TEST_F() 92 module->AddEmbeddedComputation(CreateMapComputation(negate_computation)); in TEST_F() 94 module->AddEmbeddedComputation(CreateMapComputation(negate_computation)); in TEST_F()
|
D | liveness_util_test.cc | 402 module_->AddEmbeddedComputation(make_cond()); in TEST_F() 404 module_->AddEmbeddedComputation(make_body()); in TEST_F() 439 auto sub_computation = module_->AddEmbeddedComputation(sub_builder.Build()); in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | while_transformer_test.cc | 148 module_->AddEmbeddedComputation(BuildConditionComputation(0, 10)); in TEST_F() 149 auto body = module_->AddEmbeddedComputation(BuildBodyComputation(0, 1, 1)); in TEST_F() 168 module_->AddEmbeddedComputation(BuildConditionComputation(1, 10)); in TEST_F() 169 auto body = module_->AddEmbeddedComputation(BuildBodyComputation(1, 0, 1)); in TEST_F() 188 module_->AddEmbeddedComputation(BuildConditionComputation(0, 5)); in TEST_F() 189 auto body = module_->AddEmbeddedComputation(BuildBodyComputation(0, 1, 1)); in TEST_F() 207 module_->AddEmbeddedComputation(BuildConditionComputation(0, 10)); in TEST_F() 208 auto body = module_->AddEmbeddedComputation(BuildBodyComputation(0, 1, -1)); in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | cpu_copy_insertion_test.cc | 78 HloComputation* body = module->AddEmbeddedComputation(body_builder.Build()); in TEST_F() 86 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F() 123 module->AddEmbeddedComputation(sub_builder.Build()); in TEST_F()
|