Home
last modified time | relevance | path

Searched defs:fusion_computation (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dparallel_reduction_test.cc97 HloComputation* fusion_computation; in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/
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
Dbfloat16_propagation.cc751 auto fusion_computation = hlo->fused_instructions_computation(); in ResolveInconsistentFusions() local
Dmemory_space_assignment_test.cc1039 HloComputation* fusion_computation = in TEST_P() local
1190 HloComputation* fusion_computation = in TEST_P() local
3225 HloComputation* fusion_computation = in TEST_P() local
3264 HloComputation* fusion_computation = in TEST_P() local
5038 HloComputation* fusion_computation = in TEST_P() local
Dhlo_parser.cc2072 optional<HloComputation*> fusion_computation; in ParseInstructionRhs() local
Dhlo_instruction.cc1493 HloComputation* fusion_computation) { in CreateFusion()
Dhlo_instructions.cc1380 HloComputation* fusion_computation) in HloFusionInstruction()