Searched defs:reduce_computation (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_creation_utils.cc | 387 HloComputation* reduce_computation, in MakeReduceHlo()
404 HloComputation* reduce_computation; in MakeReduceHlo() local
430 HloComputation* reduce_computation; in MakeReduceHlo() local
448 absl::Span<const int64_t> dimensions, HloComputation* reduce_computation, in MakeReduceHlo()
|
| H A D | hlo_instruction.cc | 1280 HloComputation* reduce_computation, in CreateAllReduce()
1291 HloComputation* reduce_computation, in CreateReduceScatter()
1303 HloComputation* reduce_computation, in CreateAllReduceStart()
1564 HloComputation* reduce_computation) { in CreateReduce()
1574 HloComputation* reduce_computation) { in CreateReduce()
1587 HloComputation* reduce_computation) { in CreateReduce()
1609 const Window& window, HloComputation* reduce_computation) { in CreateReduceWindow()
1617 HloComputation* reduce_computation) { in CreateReduceWindow()
|
| H A D | hlo_instructions.cc | 836 HloComputation* reduce_computation, in HloAllReduceInstructionBase()
898 HloComputation* reduce_computation, in HloReduceScatterInstruction()
1150 HloComputation* reduce_computation) in HloReduceInstruction()
2531 const Window& window, HloComputation* reduce_computation) in HloReduceWindowInstruction()
2539 HloComputation* reduce_computation) in HloReduceWindowInstruction()
|
| H A D | bfloat16_normalization_test.cc | 198 auto reduce_computation = in TEST_F() local
|
| H A D | hlo_parser.cc | 1949 optional<HloComputation*> reduce_computation; in CreateInstruction() local
2183 optional<HloComputation*> reduce_computation; in CreateInstruction() local
|
| H A D | buffer_assignment_test.cc | 991 auto reduce_computation = in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| H A D | gpu_ldg_test.cc | 115 HloComputation* reduce_computation; in TEST_F() local
|
| H A D | parallel_reduction_test.cc | 121 HloComputation* reduce_computation; in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | horizontal_input_fusion_test.cc | 83 HloComputation* reduce_computation; in TEST_F() local
|