Home
last modified time | relevance | path

Searched refs:fusion_computation (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
H A Dparallel_reduction_test.cc138 HloComputation* fusion_computation; in TEST_F() local
153 fusion_computation = module->AddEmbeddedComputation(fusion_builder.Build()); in TEST_F()
169 HloInstruction::FusionKind::kInput, entry_params, fusion_computation)); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dbfloat16_propagation.cc745 auto fusion_computation = hlo->fused_instructions_computation(); in ResolveInconsistentFusions() local
746 auto fusion_root = fusion_computation->root_instruction(); in ResolveInconsistentFusions()
755 fusion_computation->DeepCopyInstructionWithCustomCopier( in ResolveInconsistentFusions()
767 fusion_computation->set_root_instruction(copy); in ResolveInconsistentFusions()
H A Dheap_simulator_test.cc526 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()
H A Dmemory_space_assignment_test.cc1074 HloComputation* fusion_computation = in TEST_P() local
1096 shape, HloInstruction::FusionKind::kCustom, {tuple}, fusion_computation)); in TEST_P()
1225 HloComputation* fusion_computation = in TEST_P() local
1245 shape, HloInstruction::FusionKind::kCustom, {tuple}, fusion_computation)); in TEST_P()
3186 HloComputation* fusion_computation = in TEST_P() local
3193 fusion_computation)); in TEST_P()
3225 HloComputation* fusion_computation = in TEST_P() local
3239 shape, HloInstruction::FusionKind::kCustom, {tuple}, fusion_computation)); in TEST_P()
6384 HloComputation* fusion_computation = in TEST_P() local
6393 fusion_computation)); in TEST_P()
[all …]
H A Dhlo_instructions.cc1827 HloComputation* fusion_computation, absl::string_view prefix) in HloFusionInstruction() argument
1829 fusion_computation, prefix), in HloFusionInstruction()
1831 fusion_computation->SetFusionInstruction(this); in HloFusionInstruction()
H A Dhlo_parser.cc2422 optional<HloComputation*> fusion_computation; in CreateInstruction() local
2424 &fusion_computation}; in CreateInstruction()
2432 *shape, *fusion_kind, operands, *fusion_computation)); in CreateInstruction()
H A Dhlo_instruction.h1096 HloComputation* fusion_computation, absl::string_view prefix = "");
H A Dhlo_instructions.h1214 HloComputation* fusion_computation,
H A Dhlo_instruction.cc1791 HloComputation* fusion_computation, absl::string_view prefix) { in CreateFusion() argument
1793 fusion_computation, prefix); in CreateFusion()