| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | parallel_reduction_test.cc | 167 b.AddInstruction(HloInstruction::CreateFusion( in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | multi_output_fusion.cc | 163 HloInstruction* MultiOutputFusion::CreateFusion(HloInstruction* base, in CreateFusion() function in xla::MultiOutputFusion
166 computation()->AddInstruction(HloInstruction::CreateFusion( in CreateFusion()
|
| D | multi_output_fusion.h | 127 HloInstruction* CreateFusion(HloInstruction* base, HloInstruction* to_fuse);
|
| D | bfloat16_propagation_test.cc | 399 auto fusion0 = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_F()
412 auto fusion1 = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_F()
444 auto fusion = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_F()
478 auto fusion = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_F()
517 auto fusion = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_F()
648 auto body_fusion = builder_body.AddInstruction(HloInstruction::CreateFusion( in TEST_F()
|
| D | hlo_reachability_test.cc | 232 auto* fusion = computation->AddInstruction(HloInstruction::CreateFusion( in TEST_F()
|
| D | memory_space_assignment_test.cc | 1095 HloInstruction* fusion = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_P()
1244 HloInstruction* fusion = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_P()
3191 HloInstruction* fusion = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_P()
3238 HloInstruction* fusion = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_P()
3280 HloInstruction* fusion0 = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_P()
3306 HloInstruction::CreateFusion(shape, HloInstruction::FusionKind::kCustom, in TEST_P()
3364 HloInstruction* fusion0 = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_P()
3382 HloInstruction::CreateFusion(shape, HloInstruction::FusionKind::kCustom, in TEST_P()
3439 HloInstruction* fusion0 = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_P()
3443 HloInstruction::CreateFusion(shape, HloInstruction::FusionKind::kCustom, in TEST_P()
[all …]
|
| D | hlo_module_test.cc | 157 HloInstruction::CreateFusion(r0f32_, HloInstruction::FusionKind::kInput, in TEST_F()
|
| D | heap_simulator_test.cc | 529 auto fusion = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_F()
603 auto fusion = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_F()
656 auto fusion = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_F()
|
| D | hlo_instruction.h | 1090 static std::unique_ptr<HloInstruction> CreateFusion(
1093 static std::unique_ptr<HloInstruction> CreateFusion(
|
| D | instruction_fusion.cc | 717 HloInstruction::CreateFusion(consumer->shape(), kind, consumer)); in AddFusionInstruction()
|
| D | hlo_creation_utils.cc | 690 HloInstruction::CreateFusion(fused->shape(), kind, fused)); in MakeFusionInstruction()
|
| D | hlo_computation.cc | 743 HloInstruction::CreateFusion(root->shape(), fusion_kind, root)); in CreateFusionInstruction()
|
| D | hlo_instruction.cc | 423 CreateFusion(shape, fusion_kind, all_operands(), fused_computation); in CreateFromProto()
1783 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateFusion( in CreateFusion() function in xla::HloInstruction
1788 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateFusion( in CreateFusion() function in xla::HloInstruction
|
| D | hlo_parser.cc | 2431 return builder->AddInstruction(HloInstruction::CreateFusion( in CreateInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | fusion_merger_test.cc | 351 return HloInstruction::CreateFusion( in TEST_F()
|
| D | fusion_merger.cc | 184 consumer = computation_->AddInstruction(HloInstruction::CreateFusion( in FuseIntoAllUsers()
|
| D | multi_output_fusion.cc | 337 computation_->AddInstruction(HloInstruction::CreateFusion( in DoMultiOutputFusion()
|
| D | multi_output_fusion_test.cc | 749 return HloInstruction::CreateFusion( in TEST_F()
|
| D | horizontal_loop_fusion.cc | 455 HloInstruction::CreateFusion(fused_comp->root_instruction()->shape(), in Fuse()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment_test.cc | 300 module->entry_computation()->AddInstruction(HloInstruction::CreateFusion( in RunDotOutputFusion()
|