Searched refs:lhs_element (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 633 llvm::Value* lhs_element = lhs_array_.EmitReadArrayElement(lhs_index, b_); in EmitNaiveLlvmIrGemm() local
642 b_->CreateFSub(b_->CreateFMul(real(lhs_element), real(rhs_element)), in EmitNaiveLlvmIrGemm()
643 b_->CreateFMul(imag(lhs_element), imag(rhs_element))); in EmitNaiveLlvmIrGemm()
645 b_->CreateFAdd(b_->CreateFMul(real(lhs_element), imag(rhs_element)), in EmitNaiveLlvmIrGemm()
646 b_->CreateFMul(imag(lhs_element), real(rhs_element))); in EmitNaiveLlvmIrGemm()
652 llvm::Value* product = b_->CreateMul(lhs_element, rhs_element); in EmitNaiveLlvmIrGemm()
655 llvm::Value* product = b_->CreateAnd(lhs_element, rhs_element); in EmitNaiveLlvmIrGemm()
658 llvm::Value* product = b_->CreateFMul(lhs_element, rhs_element); in EmitNaiveLlvmIrGemm()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorContractionCuda.h | 325 #define lhs_element(i, j) lhs_block[72 * ((i) + 8 * (j))] in EigenContractionKernelInternal() macro
329 lcol(0) = lhs_element(0, j); \ in EigenContractionKernelInternal()
331 lcol(1) = lhs_element(1, j); \ in EigenContractionKernelInternal()
333 lcol(2) = lhs_element(2, j); \ in EigenContractionKernelInternal()
335 lcol(3) = lhs_element(3, j); \ in EigenContractionKernelInternal()
337 lcol(4) = lhs_element(4, j); \ in EigenContractionKernelInternal()
339 lcol(5) = lhs_element(5, j); \ in EigenContractionKernelInternal()
341 lcol(6) = lhs_element(6, j); \ in EigenContractionKernelInternal()
343 lcol(7) = lhs_element(7, j); \ in EigenContractionKernelInternal()
379 #undef lhs_element in EigenContractionKernelInternal()
|