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.cc746 auto fusion_computation = hlo->fused_instructions_computation(); in ResolveInconsistentFusions() local
Dmemory_space_assignment_test.cc1058 HloComputation* fusion_computation = in TEST_P() local
1209 HloComputation* fusion_computation = in TEST_P() local
3244 HloComputation* fusion_computation = in TEST_P() local
3283 HloComputation* fusion_computation = in TEST_P() local
5829 HloComputation* fusion_computation = in TEST_P() local
Dhlo_parser.cc2209 optional<HloComputation*> fusion_computation; in ParseInstructionRhs() local
Dhlo_instruction.cc1670 HloComputation* fusion_computation) { in CreateFusion()
Dhlo_instructions.cc1494 HloComputation* fusion_computation) in HloFusionInstruction()