| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_creation_utils.cc | 387 HloComputation* reduce_computation, in MakeReduceHlo() argument
394 dimensions, reduce_computation), in MakeReduceHlo()
404 HloComputation* reduce_computation; in MakeReduceHlo() local
413 reduce_computation = in MakeReduceHlo()
416 return MakeReduceHlo(operand, init_value, dimensions, reduce_computation, in MakeReduceHlo()
430 HloComputation* reduce_computation; in MakeReduceHlo() local
439 reduce_computation = module->AddEmbeddedComputation(b.Build()); in MakeReduceHlo()
441 return MakeReduceHlo(operand, init_value, all_dims, reduce_computation, in MakeReduceHlo()
448 absl::Span<const int64_t> dimensions, HloComputation* reduce_computation, in MakeReduceHlo() argument
452 auto root = reduce_computation->root_instruction(); in MakeReduceHlo()
[all …]
|
| H A D | hlo_instruction.h | 717 HloComputation* reduce_computation,
726 HloComputation* reduce_computation,
746 HloComputation* reduce_computation,
916 HloComputation* reduce_computation);
932 HloComputation* reduce_computation);
941 HloComputation* reduce_computation);
948 const Window& window, HloComputation* reduce_computation);
958 HloComputation* reduce_computation);
|
| H A D | hlo_instruction.cc | 1280 HloComputation* reduce_computation, in CreateAllReduce() argument
1284 HloOpcode::kAllReduce, shape, operands, reduce_computation, in CreateAllReduce()
1291 HloComputation* reduce_computation, in CreateReduceScatter() argument
1296 shape, operands, reduce_computation, replica_groups, constrain_layout, in CreateReduceScatter()
1303 HloComputation* reduce_computation, in CreateAllReduceStart() argument
1307 HloOpcode::kAllReduceStart, shape, operands, reduce_computation, in CreateAllReduceStart()
1564 HloComputation* reduce_computation) { in CreateReduce() argument
1566 shape, {operand, init_value}, dimensions_to_reduce, reduce_computation)); in CreateReduce()
1574 HloComputation* reduce_computation) { in CreateReduce() argument
1580 shape, all_args, dimensions_to_reduce, reduce_computation); in CreateReduce()
[all …]
|
| H A D | hlo_creation_utils.h | 192 HloComputation* reduce_computation,
211 absl::Span<const int64_t> dimensions, HloComputation* reduce_computation,
|
| H A D | hlo_instructions.cc | 836 HloComputation* reduce_computation, in HloAllReduceInstructionBase() argument
842 AppendComputation(reduce_computation); in HloAllReduceInstructionBase()
898 HloComputation* reduce_computation, in HloReduceScatterInstruction() argument
903 HloOpcode::kReduceScatter, shape, operands, reduce_computation, in HloReduceScatterInstruction()
1150 HloComputation* reduce_computation) in HloReduceInstruction() argument
1156 AppendComputation(reduce_computation); in HloReduceInstruction()
2531 const Window& window, HloComputation* reduce_computation) in HloReduceWindowInstruction() argument
2534 reduce_computation) {} in HloReduceWindowInstruction()
2539 HloComputation* reduce_computation) in HloReduceWindowInstruction() argument
2547 AppendComputation(reduce_computation); in HloReduceWindowInstruction()
|
| H A D | bfloat16_normalization_test.cc | 198 auto reduce_computation = in TEST_F() local
207 f32_output_shape, input, init, {0}, reduce_computation)); in TEST_F()
|
| H A D | hlo_instructions.h | 624 HloComputation* reduce_computation,
681 HloComputation* reduce_computation,
863 HloComputation* reduce_computation);
1670 HloComputation* reduce_computation);
1674 HloComputation* reduce_computation);
|
| H A D | hlo_parser.cc | 1949 optional<HloComputation*> reduce_computation; in CreateInstruction() local
1953 &reduce_computation}; in CreateInstruction()
1969 reduce_computation.value()->ComputeProgramShape()); in CreateInstruction()
1980 *window, *reduce_computation)); in CreateInstruction()
2183 optional<HloComputation*> reduce_computation; in CreateInstruction() local
2185 &reduce_computation}; in CreateInstruction()
2206 reduce_computation.value()->ComputeProgramShape()); in CreateInstruction()
2217 *dimensions_to_reduce, *reduce_computation)); in CreateInstruction()
|
| H A D | buffer_assignment_test.cc | 991 auto reduce_computation = in TEST_F() local
1007 /*dimensions_to_reduce=*/{0}, reduce_computation)); in TEST_F()
|
| /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
124 reduce_computation = in TEST_F()
138 {0}, reduce_computation)); in TEST_F()
|
| H A D | parallel_reduction_test.cc | 121 HloComputation* reduce_computation; in TEST_F() local
132 reduce_computation = in TEST_F()
149 output_shape, param, constant, {0}, reduce_computation)); in TEST_F()
|
| /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
92 reduce_computation = in TEST_F()
127 output_shape, mul, const0, {1}, reduce_computation)); in TEST_F()
|