Searched refs:lhs_element (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 490 llvm::Value* lhs_element = lhs_array_.EmitReadArrayElement(lhs_index, b_); in EmitNaiveLlvmIrGemm() local
499 b_->CreateFSub(b_->CreateFMul(real(lhs_element), real(rhs_element)), in EmitNaiveLlvmIrGemm()
500 b_->CreateFMul(imag(lhs_element), imag(rhs_element))); in EmitNaiveLlvmIrGemm()
502 b_->CreateFAdd(b_->CreateFMul(real(lhs_element), imag(rhs_element)), in EmitNaiveLlvmIrGemm()
503 b_->CreateFMul(imag(lhs_element), real(rhs_element))); in EmitNaiveLlvmIrGemm()
509 llvm::Value* product = b_->CreateFMul(lhs_element, rhs_element); in EmitNaiveLlvmIrGemm()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 581 llvm::Value* lhs_element = lhs_array.EmitReadArrayElement(lhs_index, &b_); in HandleDot() local
588 auto value = MultiplyComplex(lhs_element, rhs_element, &b_); in HandleDot()
596 llvm::Value* product = FMul(lhs_element, rhs_element); in HandleDot()
600 llvm::Value* product = Mul(lhs_element, rhs_element); in HandleDot()
|
| /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()
|