Home
last modified time | relevance | path

Searched defs:fusion_computation (Results 1 – 7 of 7) 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
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dheap_simulator_test.cc526 auto fusion_computation = in TEST_F() local
600 auto fusion_computation = in TEST_F() local
650 auto fusion_computation = in TEST_F() local
H A Dmemory_space_assignment_test.cc1074 HloComputation* fusion_computation = in TEST_P() local
1225 HloComputation* fusion_computation = in TEST_P() local
3186 HloComputation* fusion_computation = in TEST_P() local
3225 HloComputation* fusion_computation = in TEST_P() local
6384 HloComputation* fusion_computation = in TEST_P() local
6435 HloComputation* fusion_computation = in TEST_P() local
H A Dbfloat16_propagation.cc745 auto fusion_computation = hlo->fused_instructions_computation(); in ResolveInconsistentFusions() local
H A Dhlo_parser.cc2422 optional<HloComputation*> fusion_computation; in CreateInstruction() local
H A Dhlo_instructions.cc1827 HloComputation* fusion_computation, absl::string_view prefix) in HloFusionInstruction()
H A Dhlo_instruction.cc1791 HloComputation* fusion_computation, absl::string_view prefix) { in CreateFusion()