Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.h189 llvm::ArrayRef<llvm::Type*> accumulator_types, bool is_variadic);
Delemental_ir_emitter.cc2724 std::vector<llvm::Type*> accumulator_types; in EmitElementalReduce() local
2732 accumulator_types.push_back(accumulator_llvm_type); in EmitElementalReduce()
2795 return EmitAccumResult(accumulator_addrs, accumulator_types, is_variadic); in EmitElementalReduce()
2800 llvm::ArrayRef<llvm::Type*> accumulator_types, bool is_variadic) { in EmitAccumResult() argument
2801 TF_RET_CHECK(accumulator_addrs.size() == accumulator_types.size()); in EmitAccumResult()
2805 llvm::StructType::get(b()->getContext(), accumulator_types)); in EmitAccumResult()