| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | parallel_reduction_test.cc | 124 b.AddInstruction(HloInstruction::CreateFusion( in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | multi_output_fusion.cc | 160 HloInstruction* MultiOutputFusion::CreateFusion(HloInstruction* base, in CreateFusion() function in xla::MultiOutputFusion
163 computation()->AddInstruction(HloInstruction::CreateFusion( in CreateFusion()
|
| D | multi_output_fusion.h | 123 HloInstruction* CreateFusion(HloInstruction* base, HloInstruction* to_fuse);
|
| D | bfloat16_propagation_test.cc | 398 auto fusion0 = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_F()
411 auto fusion1 = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_F()
443 auto fusion = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_F()
477 auto fusion = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_F()
516 auto fusion = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_F()
691 auto body_fusion = builder_body.AddInstruction(HloInstruction::CreateFusion( in TEST_F()
|
| D | hlo_reachability_test.cc | 229 auto* fusion = computation->AddInstruction(HloInstruction::CreateFusion( in TEST_F()
|
| D | hlo_module_test.cc | 133 HloInstruction::CreateFusion(r0f32_, HloInstruction::FusionKind::kInput, in TEST_F()
|
| D | memory_space_assignment_test.cc | 1060 HloInstruction* fusion = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_P()
1209 HloInstruction* fusion = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_P()
3230 HloInstruction* fusion = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_P()
3277 HloInstruction* fusion = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_P()
3319 HloInstruction* fusion0 = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_P()
3345 HloInstruction::CreateFusion(shape, HloInstruction::FusionKind::kCustom, in TEST_P()
3403 HloInstruction* fusion0 = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_P()
3421 HloInstruction::CreateFusion(shape, HloInstruction::FusionKind::kCustom, in TEST_P()
3478 HloInstruction* fusion0 = builder.AddInstruction(HloInstruction::CreateFusion( in TEST_P()
3482 HloInstruction::CreateFusion(shape, HloInstruction::FusionKind::kCustom, in TEST_P()
[all …]
|
| 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 | instruction_fusion.cc | 631 HloInstruction::CreateFusion(consumer->shape(), kind, consumer)); in AddFusionInstruction()
|
| D | hlo_instruction.h | 967 static std::unique_ptr<HloInstruction> CreateFusion(
970 static std::unique_ptr<HloInstruction> CreateFusion(
|
| D | hlo_computation.cc | 747 HloInstruction::CreateFusion(root->shape(), fusion_kind, root)); in CreateFusionInstruction()
|
| D | hlo_instruction.cc | 366 CreateFusion(shape, fusion_kind, all_operands(), fused_computation); in CreateFromProto()
1484 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateFusion( in CreateFusion() function in xla::HloInstruction
1490 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateFusion( in CreateFusion() function in xla::HloInstruction
|
| D | hlo_parser.cc | 2080 instruction = builder->AddInstruction(HloInstruction::CreateFusion( in ParseInstructionRhs()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | fusion_merger_test.cc | 351 return HloInstruction::CreateFusion( in TEST_F()
|
| D | fusion_merger.cc | 300 computation_->AddInstruction(HloInstruction::CreateFusion( in HandleFusion()
|
| D | multi_output_fusion.cc | 311 computation_->AddInstruction(HloInstruction::CreateFusion( in DoMultiOutputFusion()
|
| D | multi_output_fusion_test.cc | 749 return HloInstruction::CreateFusion( in TEST_F()
|
| D | horizontal_loop_fusion.cc | 436 computation_->AddInstruction(HloInstruction::CreateFusion( in Fuse()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment_test.cc | 301 module->entry_computation()->AddInstruction(HloInstruction::CreateFusion( in RunDotOutputFusion()
|