Searched refs:ReductionOutputMap (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | ir_emitter_unnested.h | 151 using ReductionOutputMap = variable
368 Status EmitExtraOutputsForReduce(const ReductionOutputMap& result_ir_arrays,
613 const ReductionOutputMap& result_ir_arrays,
622 const IrEmitterUnnested::ReductionOutputMap& output_arrays,
630 llvm::Type* index_ty, const ReductionOutputMap& output_arrays,
637 llvm::Type* index_ty, const ReductionOutputMap& output_arrays,
644 const ReductionOutputMap& result_ir_arrays,
655 const ReductionOutputMap& result_ir_arrays);
|
| H A D | ir_emitter_unnested.cc | 1866 const ReductionOutputMap& result_ir_arrays, const IrArray::Index& index, in EmitExtraOutputsForReduce()
3724 const IrEmitterUnnested::ReductionOutputMap& output_arrays, in GetOutputAddressForReduction()
3837 const ReductionOutputMap& output_arrays, in EmitReductionOutputForRowReduction()
3964 const ReductionOutputMap& output_arrays, in EmitReductionOutputForColumnReduction()
4644 const ReductionOutputMap& result_ir_arrays) { in GenerateElementForReducer()
4694 FusedIrEmitter& fused_emitter, const ReductionOutputMap& result_ir_arrays, in EmitIRForReduction()
4960 ReductionOutputMap result_ir_arrays; in EmitUnnestedReduction()
|