Searched refs:fusion_computation (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | parallel_reduction_test.cc | 97 HloComputation* fusion_computation; in TEST_F() local
112 fusion_computation = module->AddEmbeddedComputation(fusion_builder.Build()); in TEST_F()
126 HloInstruction::FusionKind::kInput, entry_params, fusion_computation)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_propagation.cc | 746 auto fusion_computation = hlo->fused_instructions_computation(); in ResolveInconsistentFusions() local
747 auto fusion_root = fusion_computation->root_instruction(); in ResolveInconsistentFusions()
756 fusion_computation->DeepCopyInstructionWithCustomCopier( in ResolveInconsistentFusions()
768 fusion_computation->set_root_instruction(copy); in ResolveInconsistentFusions()
|
| D | heap_simulator_test.cc | 526 auto fusion_computation = in TEST_F() local
531 HloInstruction::FusionKind::kLoop, {negate}, fusion_computation)); in TEST_F()
600 auto fusion_computation = in TEST_F() local
605 HloInstruction::FusionKind::kLoop, {negate}, fusion_computation)); in TEST_F()
650 auto fusion_computation = in TEST_F() local
658 HloInstruction::FusionKind::kLoop, {neg}, fusion_computation)); in TEST_F()
|
| D | memory_space_assignment_test.cc | 1058 HloComputation* fusion_computation = in TEST_P() local
1080 shape, HloInstruction::FusionKind::kCustom, {tuple}, fusion_computation)); in TEST_P()
1209 HloComputation* fusion_computation = in TEST_P() local
1229 shape, HloInstruction::FusionKind::kCustom, {tuple}, fusion_computation)); in TEST_P()
3244 HloComputation* fusion_computation = in TEST_P() local
3251 fusion_computation)); in TEST_P()
3283 HloComputation* fusion_computation = in TEST_P() local
3297 shape, HloInstruction::FusionKind::kCustom, {tuple}, fusion_computation)); in TEST_P()
5829 HloComputation* fusion_computation = in TEST_P() local
5840 fusion_computation)); in TEST_P()
|
| D | hlo_instructions.cc | 1494 HloComputation* fusion_computation) in HloFusionInstruction() argument
1500 AppendComputation(fusion_computation); in HloFusionInstruction()
1501 fusion_computation->SetFusionInstruction(this); in HloFusionInstruction()
|
| D | hlo_instruction.h | 1073 HloComputation* fusion_computation);
|
| D | hlo_parser.cc | 2209 optional<HloComputation*> fusion_computation; in ParseInstructionRhs() local
2211 &fusion_computation}; in ParseInstructionRhs()
2218 shape, *fusion_kind, operands, *fusion_computation)); in ParseInstructionRhs()
|
| D | hlo_instructions.h | 955 HloComputation* fusion_computation);
|
| D | hlo_instruction.cc | 1670 HloComputation* fusion_computation) { in CreateFusion() argument
1672 fusion_computation); in CreateFusion()
|