Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc1545 for (llvm::Value* accumulator_shard : accumulator) { in EmitInnerLoopForVectorizedReduction()
1547 auto shard_type = accumulator_shard->getType()->getPointerElementType(); in EmitInnerLoopForVectorizedReduction()
1555 AlignedStore(initial_value, accumulator_shard, element_alignment); in EmitInnerLoopForVectorizedReduction()
1603 for (auto accumulator_shard : accumulator) { in EmitInnerLoopForVectorizedReduction() local
1604 result_ssa.push_back(AlignedLoad(accumulator_shard, element_alignment)); in EmitInnerLoopForVectorizedReduction()