Home
last modified time | relevance | path

Searched refs:lhs_array_ (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc224 const llvm_ir::IrArray& lhs_array_; member in xla::cpu::__anonb70c17dc0111::DotOpEmitter
245 lhs_array_(lhs_array), in DotOpEmitter()
256 llvm::Value* operand_ptrs[] = {lhs_array_.GetBasePointer(), in EmitLinalgMatmul()
346 llvm::Value* lhs = lhs_array_.GetBasePointer(); in EmitTiledLlvmIrGemm()
455 swap_operands ? rhs_array_.GetBasePointer() : lhs_array_.GetBasePointer(); in EmitTiledLlvmIrGemv()
457 swap_operands ? lhs_array_.GetBasePointer() : rhs_array_.GetBasePointer(); in EmitTiledLlvmIrGemv()
517 const Shape& lhs_shape = lhs_array_.GetShape(); in Emit()
552 const Shape& lhs_shape = lhs_array_.GetShape(); in EmitNaiveLlvmIrGemm()
577 lhs_array_, /*dimension_to_skip=*/lhs_reduction_dimension, "lhs"); in EmitNaiveLlvmIrGemm()
633 llvm::Value* lhs_element = lhs_array_.EmitReadArrayElement(lhs_index, b_); in EmitNaiveLlvmIrGemm()
[all …]