Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc2723 std::vector<llvm::Value*> accumulator_addrs; in EmitElementalReduce() local
2741 accumulator_addrs.push_back(accumulator_addr); in EmitElementalReduce()
2774 for (llvm::Value* accum : accumulator_addrs) { in EmitElementalReduce()
2792 Store(results[i], accumulator_addrs[i]); in EmitElementalReduce()
2795 return EmitAccumResult(accumulator_addrs, accumulator_types, is_variadic); in EmitElementalReduce()
2799 absl::Span<llvm::Value* const> accumulator_addrs, in EmitAccumResult() argument
2801 TF_RET_CHECK(accumulator_addrs.size() == accumulator_types.size()); in EmitAccumResult()
2806 for (int64 i = 0; i < accumulator_addrs.size(); i++) { in EmitAccumResult()
2807 llvm::Value* accumulator_value = Load(accumulator_addrs[i]); in EmitAccumResult()
2813 CHECK_EQ(accumulator_addrs.size(), 1); in EmitAccumResult()
[all …]
Delemental_ir_emitter.h188 absl::Span<llvm::Value* const> accumulator_addrs,