Searched refs:reduce_computation (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | gpu_ldg_test.cc | 101 HloComputation* reduce_computation; in TEST_F() local
110 reduce_computation = in TEST_F()
124 {0}, reduce_computation)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.cc | 259 HloComputation* reduce_computation; in MakeReduceHlo() local
268 reduce_computation = module->AddEmbeddedComputation(b.Build()); in MakeReduceHlo()
272 scalar_shape, operand, init_value, all_dims, reduce_computation)); in MakeReduceHlo()
|
| D | hlo_instruction.h | 489 HloComputation* reduce_computation,
611 HloComputation* reduce_computation);
627 HloComputation* reduce_computation);
634 const Window& window, HloComputation* reduce_computation);
|
| D | bfloat16_normalization_test.cc | 196 auto reduce_computation = in TEST_F() local
205 f32_output_shape, input, init, {0}, reduce_computation)); in TEST_F()
|
| D | hlo_instruction.cc | 834 HloComputation* reduce_computation, in CreateAllReduce() argument
838 shape, operands, reduce_computation, replica_groups, barrier, in CreateAllReduce()
1035 HloComputation* reduce_computation) { in CreateReduce() argument
1037 shape, {operand, init_value}, dimensions_to_reduce, reduce_computation)); in CreateReduce()
1045 HloComputation* reduce_computation) { in CreateReduce() argument
1051 shape, all_args, dimensions_to_reduce, reduce_computation); in CreateReduce()
1056 const Window& window, HloComputation* reduce_computation) { in CreateReduceWindow() argument
1058 shape, operand, init_value, window, reduce_computation); in CreateReduceWindow()
|
| D | hlo_instructions.cc | 520 HloComputation* reduce_computation, in HloAllReduceInstruction() argument
527 AppendComputation(reduce_computation); in HloAllReduceInstruction()
730 HloComputation* reduce_computation) in HloReduceInstruction() argument
736 AppendComputation(reduce_computation); in HloReduceInstruction()
1954 const Window& window, HloComputation* reduce_computation) in HloReduceWindowInstruction() argument
1958 AppendComputation(reduce_computation); in HloReduceWindowInstruction()
|
| D | hlo_instructions.h | 316 HloComputation* reduce_computation,
457 HloComputation* reduce_computation);
1099 HloComputation* reduce_computation);
|
| D | hlo_parser.cc | 1047 optional<HloComputation*> reduce_computation; in ParseInstructionRhs() local
1051 &reduce_computation}; in ParseInstructionRhs()
1061 *reduce_computation)); in ParseInstructionRhs()
1196 optional<HloComputation*> reduce_computation; in ParseInstructionRhs() local
1198 &reduce_computation}; in ParseInstructionRhs()
1216 *dimensions_to_reduce, *reduce_computation)); in ParseInstructionRhs()
|
| D | buffer_assignment_test.cc | 800 auto reduce_computation = in TEST_F() local
816 /*dimensions_to_reduce=*/{0}, reduce_computation)); in TEST_F()
|