Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc491 llvm::Value* rhs_element = rhs_array_.EmitReadArrayElement(rhs_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()
Dtiled_dot_emitter.cc362 llvm::Value* rhs_element = vsl_.LoadScalar(rhs_, col); in EmitInnerLoopEpilogue() local
370 vsl_.LoadScalar(lhs_base_pointer, scalar_row), rhs_element); in EmitInnerLoopEpilogue()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc584 llvm::Value* rhs_element = rhs_array.EmitReadArrayElement(rhs_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/
DTensorContractionCuda.h326 #define rhs_element(i, j) rhs_block[72 * ((i) + 8 * (j))] in EigenContractionKernelInternal() macro
330 rrow(0) = rhs_element(i, 0); \ in EigenContractionKernelInternal()
332 rrow(1) = rhs_element(i, 1); \ in EigenContractionKernelInternal()
334 rrow(2) = rhs_element(i, 2); \ in EigenContractionKernelInternal()
336 rrow(3) = rhs_element(i, 3); \ in EigenContractionKernelInternal()
338 rrow(4) = rhs_element(i, 4); \ in EigenContractionKernelInternal()
340 rrow(5) = rhs_element(i, 5); \ in EigenContractionKernelInternal()
342 rrow(6) = rhs_element(i, 6); \ in EigenContractionKernelInternal()
344 rrow(7) = rhs_element(i, 7); \ in EigenContractionKernelInternal()
380 #undef rhs_element in EigenContractionKernelInternal()