HomeSort by relevance Sort by last modified time
    Searched refs:lhs_element (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
dot_op_emitter.cc 490 llvm::Value* lhs_element = lhs_array_.EmitReadArrayElement(lhs_index, b_); local
499 b_->CreateFSub(b_->CreateFMul(real(lhs_element), real(rhs_element)),
500 b_->CreateFMul(imag(lhs_element), imag(rhs_element)));
502 b_->CreateFAdd(b_->CreateFMul(real(lhs_element), imag(rhs_element)),
503 b_->CreateFMul(imag(lhs_element), real(rhs_element)));
509 llvm::Value* product = b_->CreateFMul(lhs_element, rhs_element);
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
ir_emitter.cc 574 // updated_accum = accum + lhs_element * rhs_element
581 llvm::Value* lhs_element = lhs_array.EmitReadArrayElement(lhs_index, &b_); local
588 auto value = MultiplyComplex(lhs_element, rhs_element, &b_);
596 llvm::Value* product = FMul(lhs_element, rhs_element);
600 llvm::Value* product = Mul(lhs_element, rhs_element);
    [all...]
  /external/eigen/unsupported/Eigen/CXX11/src/Tensor/
TensorContractionCuda.h 325 #define lhs_element(i, j) lhs_block[72 * ((i) + 8 * (j))] macro
329 lcol(0) = lhs_element(0, j); \
331 lcol(1) = lhs_element(1, j); \
333 lcol(2) = lhs_element(2, j); \
335 lcol(3) = lhs_element(3, j); \
337 lcol(4) = lhs_element(4, j); \
339 lcol(5) = lhs_element(5, j); \
341 lcol(6) = lhs_element(6, j); \
343 lcol(7) = lhs_element(7, j); \
379 #undef lhs_element macro
    [all...]

Completed in 408 milliseconds