| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| 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()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | parallel_reduction_test.cc | 80 HloComputation* reduce_computation; in TEST_F() local
91 reduce_computation = in TEST_F()
108 output_shape, param, constant, {0}, reduce_computation)); in TEST_F()
|
| D | gpu_ldg_test.cc | 116 HloComputation* reduce_computation; in TEST_F() local
125 reduce_computation = in TEST_F()
139 {0}, reduce_computation)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.cc | 330 HloComputation* reduce_computation; in MakeReduceHlo() local
339 reduce_computation = in MakeReduceHlo()
344 result_shape, operand, init_value, dimensions, reduce_computation)); in MakeReduceHlo()
356 HloComputation* reduce_computation; in MakeReduceHlo() local
365 reduce_computation = module->AddEmbeddedComputation(b.Build()); in MakeReduceHlo()
369 scalar_shape, operand, init_value, all_dims, reduce_computation)); in MakeReduceHlo()
|
| D | hlo_instruction.h | 653 HloComputation* reduce_computation,
810 HloComputation* reduce_computation);
826 HloComputation* reduce_computation);
833 const Window& window, HloComputation* reduce_computation);
843 HloComputation* reduce_computation);
|
| 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()
|
| D | hlo_instructions.cc | 667 HloComputation* reduce_computation, in HloAllReduceInstruction() argument
673 AppendComputation(reduce_computation); in HloAllReduceInstruction()
904 HloComputation* reduce_computation) in HloReduceInstruction() argument
910 AppendComputation(reduce_computation); in HloReduceInstruction()
2232 const Window& window, HloComputation* reduce_computation) in HloReduceWindowInstruction() argument
2235 reduce_computation) {} in HloReduceWindowInstruction()
2240 HloComputation* reduce_computation) in HloReduceWindowInstruction() argument
2248 AppendComputation(reduce_computation); in HloReduceWindowInstruction()
|
| D | hlo_instruction.cc | 1054 HloComputation* reduce_computation, in CreateAllReduce() argument
1058 shape, operands, reduce_computation, replica_groups, constrain_layout, in CreateAllReduce()
1289 HloComputation* reduce_computation) { in CreateReduce() argument
1291 shape, {operand, init_value}, dimensions_to_reduce, reduce_computation)); in CreateReduce()
1299 HloComputation* reduce_computation) { in CreateReduce() argument
1305 shape, all_args, dimensions_to_reduce, reduce_computation); in CreateReduce()
1310 const Window& window, HloComputation* reduce_computation) { in CreateReduceWindow() argument
1312 shape, operand, init_value, window, reduce_computation); in CreateReduceWindow()
1318 HloComputation* reduce_computation) { in CreateReduceWindow() argument
1320 shape, operands, init_values, window, reduce_computation); in CreateReduceWindow()
|
| D | hlo_instructions.h | 424 HloComputation* reduce_computation,
590 HloComputation* reduce_computation);
1297 HloComputation* reduce_computation);
1301 HloComputation* reduce_computation);
|
| D | hlo_parser.cc | 1571 optional<HloComputation*> reduce_computation; in ParseInstructionRhs() local
1575 &reduce_computation}; in ParseInstructionRhs()
1591 reduce_computation.value()->ComputeProgramShape()); in ParseInstructionRhs()
1603 *window, *reduce_computation)); in ParseInstructionRhs()
1816 optional<HloComputation*> reduce_computation; in ParseInstructionRhs() local
1818 &reduce_computation}; in ParseInstructionRhs()
1838 reduce_computation.value()->ComputeProgramShape()); in ParseInstructionRhs()
1850 *dimensions_to_reduce, *reduce_computation)); in ParseInstructionRhs()
|
| D | buffer_assignment_test.cc | 972 auto reduce_computation = in TEST_F() local
988 /*dimensions_to_reduce=*/{0}, reduce_computation)); in TEST_F()
|