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

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
dot_op_emitter.cc 281 llvm::Value* rhs_element = vsl_.LoadScalar(rhs_, col);
290 vsl_.LoadScalar(lhs_base_pointer, scalar_row), rhs_element);
817 llvm::Value* rhs_element local
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
ir_emitter.cc 539 // updated_accum = accum + lhs_element * rhs_element
546 llvm::Value* rhs_element = local
554 ir_builder_.CreateFMul(REAL(lhs_element), REAL(rhs_element)),
555 ir_builder_.CreateFMul(IMAG(lhs_element), IMAG(rhs_element)));
557 ir_builder_.CreateFMul(REAL(lhs_element), IMAG(rhs_element)),
558 ir_builder_.CreateFMul(IMAG(lhs_element), REAL(rhs_element)));
566 llvm::Value* product = ir_builder_.CreateFMul(lhs_element, rhs_element);
    [all...]
  /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...]

Completed in 180 milliseconds