Searched refs:reducers (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.h | 592 absl::Span<HloComputation* const> reducers,
613 absl::Span<HloComputation* const> reducers,
630 absl::Span<HloComputation* const> reducers,
|
| D | ir_emitter_unnested.cc | 4319 absl::Span<HloComputation* const> reducers, in EmitFullWarpShuffleDownLoopForAllReduces() argument
4322 CHECK_EQ(reducers.size(), partial_result_addresses.size()); in EmitFullWarpShuffleDownLoopForAllReduces()
4323 for (int i = 0; i != reducers.size(); i++) { in EmitFullWarpShuffleDownLoopForAllReduces()
4325 reducers[i], partial_result_addresses[i]->getType()->getElementType(), in EmitFullWarpShuffleDownLoopForAllReduces()
4381 absl::Span<HloComputation* const> reducers, in EmitEpilogueForReduction() argument
4481 reducers[i], element_type, current_output, in EmitEpilogueForReduction()
4511 reducers[i], element_type, in EmitEpilogueForReduction()
4517 *reducers[i], output_address, block_accum_addr)); in EmitEpilogueForReduction()
4541 reducers[i], element_type, shmem_transposed_addr, in EmitEpilogueForReduction()
4557 *reducers[i], output_address, shmem_transposed_addr)); in EmitEpilogueForReduction()
[all …]
|
| /external/python/cpython3/Doc/library/ |
| D | pickle.rst | 377 :attr:`dispatch_table`-registered reducers to pickle ``obj``.
|
| /external/jline/src/src/test/resources/jline/example/ |
| D | english.gz | |
| /external/kotlinx.coroutines/benchmarks/src/jmh/resources/ |
| D | ospd.txt.gz | |
| /external/cldr/tools/java/org/unicode/cldr/util/data/transforms/ |
| D | internal_raw_IPA-old.txt | 162358 reducers %8306
|