Searched refs:operand_ir_arrays (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.h | 590 absl::Span<const llvm_ir::IrArray> operand_ir_arrays,
603 absl::Span<const llvm_ir::IrArray> operand_ir_arrays,
622 absl::Span<const llvm_ir::IrArray> operand_ir_arrays,
|
| D | ir_emitter_unnested.cc | 4218 absl::Span<const llvm_ir::IrArray> operand_ir_arrays, in EmitPrologueForReduction() argument
4276 init_ir_value = operand_ir_arrays[1].EmitReadArrayElement( in EmitPrologueForReduction()
4598 absl::Span<const llvm_ir::IrArray> operand_ir_arrays, in EmitTileElementForReduction() argument
4628 return operand_ir_arrays[0].EmitReadArrayElement(index, &b_); in EmitTileElementForReduction()
5398 absl::Span<const llvm_ir::IrArray> operand_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()
5619 absl::Span<const llvm_ir::IrArray> operand_ir_arrays; in EmitReductionFromOrToContiguousDimensions() local
5643 operand_ir_arrays = absl::MakeSpan(ir_arrays).subspan(0, 2); in EmitReductionFromOrToContiguousDimensions()
5659 fused_emitter, operand_ir_arrays, result_ir_arrays, in EmitReductionFromOrToContiguousDimensions()
|