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

  /external/eigen/unsupported/Eigen/CXX11/src/Tensor/
TensorContractionCuda.h 326 #define rhs_element(i, j) rhs_block[72 * ((i) + 8 * (j))] macro
330 rrow(0) = rhs_element(i, 0); \
332 rrow(1) = rhs_element(i, 1); \
334 rrow(2) = rhs_element(i, 2); \
336 rrow(3) = rhs_element(i, 3); \
338 rrow(4) = rhs_element(i, 4); \
340 rrow(5) = rhs_element(i, 5); \
342 rrow(6) = rhs_element(i, 6); \
344 rrow(7) = rhs_element(i, 7); \
380 #undef rhs_element macro
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
dot_op_emitter.cc 491 llvm::Value* rhs_element = rhs_array_.EmitReadArrayElement(rhs_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
584 llvm::Value* rhs_element = rhs_array.EmitReadArrayElement(rhs_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...]

Completed in 304 milliseconds