Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_test.cc93 module->AddEmbeddedComputation(CreateConstantComputation()); in TEST_F()
108 module->AddEmbeddedComputation(CreateConstantComputation()); in TEST_F()
110 module->AddEmbeddedComputation(CreateCallComputation({computation1})); in TEST_F()
112 module->AddEmbeddedComputation(CreateCallComputation({computation1})); in TEST_F()
148 fused_computation = module->AddEmbeddedComputation(b.Build()); in TEST_F()
284 module->AddEmbeddedComputation(CreateConstantComputation()); in TEST_F()
286 module->AddEmbeddedComputation(CreateCallComputation({computation1})); in TEST_F()
288 module->AddEmbeddedComputation(CreateCallComputation({computation1})); in TEST_F()
413 module->AddEmbeddedComputation(reduction->Clone()); in TEST_F()
507 new_comp = module->AddEmbeddedComputation(b.Build()); in TEST_F()
[all …]
Dhlo_subcomputation_unification_test.cc72 module->AddEmbeddedComputation(CreateR0S32IdentityComputation()); in TEST_F()
74 module->AddEmbeddedComputation(CreateR0S32IdentityComputation()); in TEST_F()
99 module->AddEmbeddedComputation(CreateR0S32AdditionComputation()); in TEST_F()
101 module->AddEmbeddedComputation(CreateR0S32AdditionComputation()); in TEST_F()
129 module->AddEmbeddedComputation(CreateR1S32AdditionComputation(r1s32_5_)); in TEST_F()
131 module->AddEmbeddedComputation(CreateR1S32AdditionComputation(r1s32_3_)); in TEST_F()
166 module->AddEmbeddedComputation(builder.Build()); in TEST_F()
Dflatten_call_graph_test.cc114 module->AddEmbeddedComputation(MakeConditionComputation()); in TEST_F()
116 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F()
117 HloComputation* b_computation = module->AddEmbeddedComputation( in TEST_F()
129 a_computation = module->AddEmbeddedComputation(builder.Build()); in TEST_F()
165 cond_computation = module->AddEmbeddedComputation(builder.Build()); in TEST_F()
207 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F()
209 HloComputation* b_computation = module->AddEmbeddedComputation( in TEST_F()
230 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F()
Dcall_graph_test.cc122 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F()
144 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F()
177 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F()
212 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F()
256 module->AddEmbeddedComputation(MakeScalarComputation(HloOpcode::kCeil)); in TEST_F()
258 module->AddEmbeddedComputation(MakeScalarComputation(HloOpcode::kFloor)); in TEST_F()
318 module->AddEmbeddedComputation(MakeConditionComputation()); in TEST_F()
320 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F()
321 HloComputation* b_computation = module->AddEmbeddedComputation( in TEST_F()
333 a_computation = module->AddEmbeddedComputation(builder.Build()); in TEST_F()
[all …]
Dcall_inliner_test.cc55 module->AddEmbeddedComputation(inner.Build()); in TEST_F()
90 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 module->AddEmbeddedComputation(empty.Build()); in TEST_F()
161 module->AddEmbeddedComputation(empty2.Build()); in TEST_F()
195 auto outfeed_computation = module->AddEmbeddedComputation(outfeeder.Build()); in TEST_F()
Dcopy_insertion_test.cc671 auto condition = module_->AddEmbeddedComputation( in BuildWhileInstructionWithCustomInit()
673 auto body = module_->AddEmbeddedComputation( in BuildWhileInstructionWithCustomInit()
705 auto condition = module_->AddEmbeddedComputation( in TEST_F()
708 module_->AddEmbeddedComputation(BuildIndependentBodyComputation()); in TEST_F()
766 module_->AddEmbeddedComputation(while_hlo->while_condition()->Clone()); in TEST_F()
768 module_->AddEmbeddedComputation(while_hlo->while_body()->Clone()); in TEST_F()
833 module_->AddEmbeddedComputation(while_hlo->while_condition()->Clone()); in TEST_F()
835 module_->AddEmbeddedComputation(while_hlo->while_body()->Clone()); in TEST_F()
892 module_->AddEmbeddedComputation(while_hlo->while_condition()->Clone()); in TEST_F()
894 module_->AddEmbeddedComputation(while_hlo->while_body()->Clone()); in TEST_F()
[all …]
Dhlo_dce_test.cc224 module->AddEmbeddedComputation(callee_builder.Build()); in TEST_F()
266 auto cond_computation = module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
281 auto body_computation = module->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
325 module->AddEmbeddedComputation(nested_callee_builder.Build()); in TEST_F()
336 module->AddEmbeddedComputation(callee_builder.Build()); in TEST_F()
375 auto reduce_subcomp = module->AddEmbeddedComputation(subcomp_builder.Build()); in TEST_F()
416 auto reduce_subcomp = module->AddEmbeddedComputation(subcomp_builder.Build()); in TEST_F()
464 module->AddEmbeddedComputation(called_subcomp_builder.Build()); in TEST_F()
477 module->AddEmbeddedComputation(dead_subcomp_builder.Build()); in TEST_F()
488 module->AddEmbeddedComputation(dead_subcomp_builder.Build()); in TEST_F()
Dwhile_loop_invariant_code_motion_test.cc57 return module->AddEmbeddedComputation(builder.Build()); in MakeAlwaysTrueComputation()
80 return m->AddEmbeddedComputation(builder.Build()); in TEST_F()
139 return m->AddEmbeddedComputation(builder.Build()); in TEST_F()
193 return m->AddEmbeddedComputation(builder.Build()); in TEST_F()
233 return m->AddEmbeddedComputation(builder.Build()); in TEST_F()
273 return m->AddEmbeddedComputation(builder.Build()); in TEST_F()
326 return m->AddEmbeddedComputation(builder.Build()); in TEST_F()
378 return m->AddEmbeddedComputation(builder.Build()); in TEST_F()
427 while_body = m->AddEmbeddedComputation(builder.Build()); in TEST_F()
451 HloComputation* result = m->AddEmbeddedComputation(builder.Build()); in TEST_F()
Dhlo_alias_analysis_test.cc377 HloComputation* body = module_->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
386 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
445 module_->AddEmbeddedComputation(subbuilder.Build()); in TEST_F()
485 module_->AddEmbeddedComputation(subbuilder.Build()); in TEST_F()
562 HloComputation* body = module_->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
571 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
655 HloComputation* body = module_->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
663 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
728 module_->AddEmbeddedComputation(build_cond_computation()); in TEST_F()
730 module_->AddEmbeddedComputation(build_cond_computation()); in TEST_F()
[all …]
Dhlo_ordering_test.cc62 module->AddEmbeddedComputation(builder_c.Build()); in TEST_F()
70 module->AddEmbeddedComputation(builder_b.Build()); in TEST_F()
76 module->AddEmbeddedComputation(builder_a.Build()); in TEST_F()
136 HloComputation* body = module->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
144 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
219 HloComputation* body = module->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
227 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
475 HloComputation* subcomputation = module->AddEmbeddedComputation( in TEST_F()
Dflatten_call_graph.cc121 clone = module->AddEmbeddedComputation(computation->Clone()); in FlattenNode()
126 clone = module->AddEmbeddedComputation(computation->Clone()); in FlattenNode()
142 module->AddEmbeddedComputation(callee->Clone()); in FlattenNode()
Dmap_inliner_test.cc62 hlo_module->AddEmbeddedComputation(std::move(max_f32)); in TEST_F()
96 hlo_module->AddEmbeddedComputation(std::move(const2_f32)); in TEST_F()
134 hlo_module->AddEmbeddedComputation(std::move(max_f32)); in TEST_F()
166 hlo_module->AddEmbeddedComputation(std::move(param_f32)); in TEST_F()
Dbuffer_assignment_test.cc812 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
835 module->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
937 module->AddEmbeddedComputation(BuildMapComputationPlus1("f32+1")); in TEST_F()
992 module->AddEmbeddedComputation(BuildReduceComputation("f32+f32")); in TEST_F()
1043 module->AddEmbeddedComputation(BuildWhileConditionComputation("if<4")); in TEST_F()
1045 module->AddEmbeddedComputation(BuildWhileBodyComputation("add-update")); in TEST_F()
1108 auto true_computation = module->AddEmbeddedComputation( in TEST_F()
1110 auto false_computation = module->AddEmbeddedComputation( in TEST_F()
1399 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F()
1407 auto call_computation = module->AddEmbeddedComputation(call_builder.Build()); in TEST_F()
[all …]
Dbfloat16_propagation_test.cc150 auto max_computation = module->AddEmbeddedComputation(sub_builder.Build()); in TEST_F()
211 auto reduction = module->AddEmbeddedComputation(rb.Build()); in TEST_F()
398 auto comp_f0 = module->AddEmbeddedComputation(builder_f0.Build()); in TEST_F()
411 auto comp_f1 = module->AddEmbeddedComputation(builder_f1.Build()); in TEST_F()
443 auto comp_f = module->AddEmbeddedComputation(builder_f.Build()); in TEST_F()
477 auto comp_f = module->AddEmbeddedComputation(builder_f.Build()); in TEST_F()
516 auto comp_f = module->AddEmbeddedComputation(builder_f.Build()); in TEST_F()
577 auto cond = module->AddEmbeddedComputation(builder_cond.Build()); in TEST_F()
584 auto body = module->AddEmbeddedComputation(builder_body.Build()); in TEST_F()
635 auto cond = module->AddEmbeddedComputation(builder_cond.Build()); in TEST_F()
[all …]
Dhlo_rematerialization_test.cc156 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
158 HloComputation* body_computation = module->AddEmbeddedComputation( in TEST_F()
192 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
194 HloComputation* body_computation = module->AddEmbeddedComputation( in TEST_F()
224 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
226 module->AddEmbeddedComputation(while_cond->Clone()); in TEST_F()
228 HloComputation* inner_computation = module->AddEmbeddedComputation( in TEST_F()
231 module->AddEmbeddedComputation(MakeRematerializableWhileComputation( in TEST_F()
358 subcomputation = module->AddEmbeddedComputation(builder.Build()); in TEST_F()
515 subcomputation = module->AddEmbeddedComputation(builder.Build()); in TEST_P()
Dhlo_dataflow_analysis_test.cc272 module_->AddEmbeddedComputation(subbuilder.Build()); in TEST_P()
335 module_->AddEmbeddedComputation(inner_builder.Build()); in TEST_P()
346 module_->AddEmbeddedComputation(outer_builder.Build()); in TEST_P()
411 HloComputation* body = module_->AddEmbeddedComputation(body_builder.Build()); in TEST_P()
420 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_P()
514 HloComputation* body = module_->AddEmbeddedComputation(body_builder.Build()); in TEST_P()
522 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_P()
581 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_P()
592 module_->AddEmbeddedComputation(level0_builder.Build()); in TEST_P()
601 module_->AddEmbeddedComputation(level1_builder.Build()); in TEST_P()
[all …]
Dwhile_util.cc50 return narrow_condition->parent()->AddEmbeddedComputation(builder.Build()); in WidenWhileCondition()
74 return narrow_body->parent()->AddEmbeddedComputation(builder.Build()); in WidenWhileBody()
267 loop_state_shape, module->AddEmbeddedComputation(std::move(cond)), in MakeCountedLoop()
268 module->AddEmbeddedComputation(std::move(body)), owned_init_tuple.get()); in MakeCountedLoop()
Dwhile_loop_simplifier.cc173 module->AddEmbeddedComputation(std::move(new_while_cond)), in RemoveDeadTupleIndices()
174 module->AddEmbeddedComputation(std::move(new_while_body)), in RemoveDeadTupleIndices()
756 module->AddEmbeddedComputation(std::move(new_while_cond)), in TryRemoveConstantParams()
757 module->AddEmbeddedComputation(std::move(new_while_body)), in TryRemoveConstantParams()
1087 module->AddEmbeddedComputation(std::move(new_while_cond)), in TryFlattenNestedTuples()
1088 module->AddEmbeddedComputation(std::move(new_while_body)), in TryFlattenNestedTuples()
1305 module->AddEmbeddedComputation(while_body->CloneWithReplacementPairs({ in TryMergeInductionVariables()
1326 module->AddEmbeddedComputation(std::move(new_while_cond)), in TryMergeInductionVariables()
1327 module->AddEmbeddedComputation(std::move(new_while_body)), in TryMergeInductionVariables()
Ddynamic_dimension_inference_test.cc71 return module_->AddEmbeddedComputation(embedded_builder.Build()); in GetAdd()
91 return module_->AddEmbeddedComputation(embedded_builder.Build()); in GetAddTuple()
102 return module_->AddEmbeddedComputation(embedded_builder.Build()); in GetGe()
760 HloComputation* body = module_->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
768 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
851 module_->AddEmbeddedComputation(true_builder.Build()); in TEST_F()
872 module_->AddEmbeddedComputation(false_builder.Build()); in TEST_F()
1141 module_->AddEmbeddedComputation(compare_builder.Build()); in TEST_F()
1179 module_->AddEmbeddedComputation(compare_builder.Build()); in TEST_F()
Dconditional_simplifier.cc108 computation->parent()->AddEmbeddedComputation(computation->Clone()); in TryRemoveUnusedConditionalOperands()
168 conditional_op->GetModule()->AddEmbeddedComputation( in ReplaceRootWithEmptyTupleIfNoUsers()
283 conditional_op->GetModule()->AddEmbeddedComputation( in RemoveUnusedTupleElements()
Dall_gather_decomposer.cc51 HloComputation* reduction = module->AddEmbeddedComputation(sum_b.Build()); in MakeBinaryAdd()
Dtuple_simplifier_test.cc241 c0 = module->AddEmbeddedComputation(builder.Build()); in TEST_F()
256 c1 = module->AddEmbeddedComputation(builder.Build()); in TEST_F()
Dhlo_computation_test.cc84 module->AddEmbeddedComputation(CreateNegateComputation()); in TEST_F()
96 module->AddEmbeddedComputation(CreateNegateComputation()); in TEST_F()
98 module->AddEmbeddedComputation(CreateMapComputation(negate_computation)); in TEST_F()
100 module->AddEmbeddedComputation(CreateMapComputation(negate_computation)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dwhile_transformer_test.cc128 module_->AddEmbeddedComputation(BuildConditionComputation(0, 10)); in TEST_F()
129 auto body = module_->AddEmbeddedComputation(BuildBodyComputation(0, 1, 1)); in TEST_F()
139 module_->AddEmbeddedComputation(BuildConditionComputation(1, 10)); in TEST_F()
140 auto body = module_->AddEmbeddedComputation(BuildBodyComputation(1, 0, 1)); in TEST_F()
150 module_->AddEmbeddedComputation(BuildConditionComputation(0, 5)); in TEST_F()
151 auto body = module_->AddEmbeddedComputation(BuildBodyComputation(0, 1, 1)); in TEST_F()
161 module_->AddEmbeddedComputation(BuildConditionComputation(0, 10)); in TEST_F()
162 auto body = module_->AddEmbeddedComputation(BuildBodyComputation(0, 1, -1)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dparallel_reduction_test.cc133 module->AddEmbeddedComputation(embedded_builder.Build()); in TEST_F()
153 fusion_computation = module->AddEmbeddedComputation(fusion_builder.Build()); in TEST_F()

123