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.cc195 const llvm_ir::IrArray& lhs_array_; member in xla::cpu::__anond13f39450111::DotOpEmitter
217 lhs_array_(lhs_array), in DotOpEmitter()
229 llvm::Value* lhs = lhs_array_.GetBasePointer(); in EmitTiledLlvmIrGemm()
315 swap_operands ? rhs_array_.GetBasePointer() : lhs_array_.GetBasePointer(); in EmitTiledLlvmIrGemv()
317 swap_operands ? lhs_array_.GetBasePointer() : rhs_array_.GetBasePointer(); in EmitTiledLlvmIrGemv()
377 const Shape& lhs_shape = lhs_array_.GetShape(); in Emit()
409 const Shape& lhs_shape = lhs_array_.GetShape(); in EmitNaiveLlvmIrGemm()
434 lhs_array_, /*dimension_to_skip=*/lhs_reduction_dimension, "lhs"); in EmitNaiveLlvmIrGemm()
490 llvm::Value* lhs_element = lhs_array_.EmitReadArrayElement(lhs_index, b_); in EmitNaiveLlvmIrGemm()
553 lhs_array_.EmitReadArrayElement(/*index=*/element_index, b_); in EmitScalarDot()
[all …]