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 D | parallel_reduction_test.cc | 138 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 D | bfloat16_propagation.cc | 745 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 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()
|
H A D | memory_space_assignment_test.cc | 1074 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 D | hlo_instructions.cc | 1827 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 D | hlo_parser.cc | 2422 optional<HloComputation*> fusion_computation; in CreateInstruction() local 2424 &fusion_computation}; in CreateInstruction() 2432 *shape, *fusion_kind, operands, *fusion_computation)); in CreateInstruction()
|
H A D | hlo_instruction.h | 1096 HloComputation* fusion_computation, absl::string_view prefix = "");
|
H A D | hlo_instructions.h | 1214 HloComputation* fusion_computation,
|
H A D | hlo_instruction.cc | 1791 HloComputation* fusion_computation, absl::string_view prefix) { in CreateFusion() argument 1793 fusion_computation, prefix); in CreateFusion()
|