Home
last modified time | relevance | path

Searched refs:scalar_accumulators (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dtiled_dot_emitter.cc491 std::vector<ScalarVariable>* scalar_accumulators);
508 std::vector<ScalarVariable> scalar_accumulators; in EmitOuterLoopBody() local
511 scalar_accumulators.emplace_back(&vsl_, vsl_.GetZeroScalar()); in EmitOuterLoopBody()
516 &scalar_accumulators); in EmitOuterLoopBody()
539 vsl_.Add(horizontal_sums[i], scalar_accumulators[i].Get()); in EmitOuterLoopBody()
582 std::vector<ScalarVariable>* scalar_accumulators) { in EmitInnerLoopEpilogue() argument
598 llvm::Value* old_value = (*scalar_accumulators)[r].Get(); in EmitInnerLoopEpilogue()
599 (*scalar_accumulators)[r].Set(vsl_.Add(old_value, product)); in EmitInnerLoopEpilogue()