Home
last modified time | relevance | path

Searched refs:AddEmbeddedComputation (Results 1 – 25 of 43) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_test.cc73 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()
Dflatten_call_graph_test.cc113 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()
Dcall_graph_test.cc119 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 …]
Dcopy_insertion_test.cc749 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 …]
Dhlo_subcomputation_unification_test.cc73 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()
Dwhile_loop_simplifier_test.cc63 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 …]
Dcall_inliner_test.cc57 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()
Dbuffer_assignment_test.cc583 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 …]
Dwhile_loop_invariant_code_motion_test.cc56 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()
Dhlo_alias_analysis_test.cc227 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 …]
Dhlo_ordering_test.cc109 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()
Dhlo_rematerialization_test.cc220 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()
Dhlo_dataflow_analysis_test.cc258 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 …]
Dhlo_dce_test.cc189 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()
Dgraphviz_example.cc53 return module->AddEmbeddedComputation(builder.Build()); in AddScalarConstantComputation()
66 return module->AddEmbeddedComputation(builder.Build()); in ScalarSumComputation()
78 return module->AddEmbeddedComputation(builder.Build()); in CallForwardingComputation()
Dinliner_test.cc63 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()
Dhlo_scheduling_test.cc54 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
61 module->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
Dflatten_call_graph.cc87 module->AddEmbeddedComputation(computation->Clone()); in FlattenNode()
102 module->AddEmbeddedComputation(callee->Clone()); in FlattenNode()
Dwhile_util.cc39 return narrow_condition->parent()->AddEmbeddedComputation(builder.Build()); in WidenWhileCondition()
64 return narrow_body->parent()->AddEmbeddedComputation(builder.Build()); in WidenWhileBody()
Dhlo_module.cc108 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()
Dbuffer_liveness_test.cc403 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()
Dhlo_computation_test.cc78 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()
Dliveness_util_test.cc402 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/
Dwhile_transformer_test.cc148 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/
Dcpu_copy_insertion_test.cc78 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()

12