Searched refs:scalar_row (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | tiled_dot_emitter.cc | 368 /*step=*/1, [&](llvm::Value* scalar_row) { in EmitInnerLoopEpilogue() argument
370 vsl_.LoadScalar(lhs_base_pointer, scalar_row), rhs_element); in EmitInnerLoopEpilogue()
379 vsl_.Add(vsl_.LoadScalar(addend_, scalar_row), in EmitInnerLoopEpilogue()
381 result_, scalar_row); in EmitInnerLoopEpilogue()
383 vsl_.StoreScalar(product, result_, scalar_row); in EmitInnerLoopEpilogue()
389 vsl_.Add(vsl_.LoadScalar(result_, scalar_row), product), in EmitInnerLoopEpilogue()
390 result_, scalar_row); in EmitInnerLoopEpilogue()
|