Searched refs:scalar_row (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | tiled_dot_emitter.cc | 371 /*step=*/1, [&](llvm::Value* scalar_row) { in EmitInnerLoopEpilogue() argument
373 vsl_.LoadScalar(lhs_base_pointer, scalar_row), rhs_element); in EmitInnerLoopEpilogue()
382 vsl_.Add(vsl_.LoadScalar(addend_, scalar_row), in EmitInnerLoopEpilogue()
384 result_, scalar_row); in EmitInnerLoopEpilogue()
386 vsl_.StoreScalar(product, result_, scalar_row); in EmitInnerLoopEpilogue()
392 vsl_.Add(vsl_.LoadScalar(result_, scalar_row), product), in EmitInnerLoopEpilogue()
393 result_, scalar_row); in EmitInnerLoopEpilogue()
|