Home
last modified time | relevance | path

Searched defs:reduce_computation (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhorizontal_input_fusion_test.cc83 HloComputation* reduce_computation; in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_ldg_test.cc116 HloComputation* reduce_computation; in TEST_F() local
Dparallel_reduction_test.cc80 HloComputation* reduce_computation; in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils.cc330 HloComputation* reduce_computation; in MakeReduceHlo() local
356 HloComputation* reduce_computation; in MakeReduceHlo() local
Dbfloat16_normalization_test.cc198 auto reduce_computation = in TEST_F() local
Dhlo_instruction.cc1054 HloComputation* reduce_computation, in CreateAllReduce()
1289 HloComputation* reduce_computation) { in CreateReduce()
1299 HloComputation* reduce_computation) { in CreateReduce()
1310 const Window& window, HloComputation* reduce_computation) { in CreateReduceWindow()
1318 HloComputation* reduce_computation) { in CreateReduceWindow()
Dhlo_instructions.cc667 HloComputation* reduce_computation, in HloAllReduceInstruction()
904 HloComputation* reduce_computation) in HloReduceInstruction()
2232 const Window& window, HloComputation* reduce_computation) in HloReduceWindowInstruction()
2240 HloComputation* reduce_computation) in HloReduceWindowInstruction()
Dhlo_parser.cc1571 optional<HloComputation*> reduce_computation; in ParseInstructionRhs() local
1816 optional<HloComputation*> reduce_computation; in ParseInstructionRhs() local
Dbuffer_assignment_test.cc972 auto reduce_computation = in TEST_F() local