Searched refs:result_ir_arrays (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.h | 366 absl::Span<const llvm_ir::IrArray> result_ir_arrays,
591 absl::Span<const llvm_ir::IrArray> result_ir_arrays,
604 absl::Span<const llvm_ir::IrArray> result_ir_arrays,
612 absl::Span<const llvm_ir::IrArray> result_ir_arrays,
623 absl::Span<const llvm_ir::IrArray> result_ir_arrays,
|
| D | ir_emitter_unnested.cc | 2118 absl::Span<const llvm_ir::IrArray> result_ir_arrays, in EmitExtraOutputsForReduce() argument
2131 result_ir_arrays[extra_output_gens[i].second].EmitWriteArrayElement( in EmitExtraOutputsForReduce()
4219 absl::Span<const llvm_ir::IrArray> result_ir_arrays, in EmitPrologueForReduction() argument
4380 absl::Span<const llvm_ir::IrArray> result_ir_arrays, in EmitEpilogueForReduction() argument
4448 auto output_array = result_ir_arrays[index]; in EmitEpilogueForReduction()
4599 absl::Span<const llvm_ir::IrArray> result_ir_arrays, in EmitTileElementForReduction() argument
4665 result_ir_arrays, input_index, in EmitTileElementForReduction()
5399 absl::Span<const llvm_ir::IrArray> result_ir_arrays, in EmitIRForReduction() argument
5433 fused_emitter, operand_ir_arrays, result_ir_arrays, in EmitIRForReduction()
5441 fused_emitter, operand_ir_arrays, result_ir_arrays, reducers, index, in EmitIRForReduction()
[all …]
|