Searched refs:result_ir_arrays (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 | 368 Status EmitExtraOutputsForReduce(const ReductionOutputMap& result_ir_arrays,
613 const ReductionOutputMap& result_ir_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() argument
1879 absl::Span<llvm_ir::IrArray const> result_ir = result_ir_arrays.at(p.first); in EmitExtraOutputsForReduce()
4644 const ReductionOutputMap& result_ir_arrays) { in GenerateElementForReducer() argument
4694 FusedIrEmitter& fused_emitter, const ReductionOutputMap& result_ir_arrays, in EmitIRForReduction() argument
4740 num_partial_results, result_ir_arrays); in EmitIRForReduction()
4746 result_ir_arrays, input_index, reduction_info, extra_output_gens)); in EmitIRForReduction()
4766 index_ty, result_ir_arrays, reduce, in EmitIRForReduction()
4770 index_ty, result_ir_arrays, in EmitIRForReduction()
4960 ReductionOutputMap result_ir_arrays; in EmitUnnestedReduction() local
4967 result_ir_arrays[root] = in EmitUnnestedReduction()
[all …]
|