Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc677 llvm::AllocaInst* accumulator_addr = in HandleReduce() local
680 Store(Load(GetBasePointer(*init_value)), accumulator_addr); in HandleReduce()
715 *function, {accumulator_addr, input_address}, accumulator_addr)); in HandleReduce()
718 return Load(accumulator_addr); in HandleReduce()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc1824 llvm::AllocaInst* accumulator_addr = llvm_ir::EmitAllocaAtFunctionEntry( in EmitElementalReduce() local
1830 Store(init_value, accumulator_addr); in EmitElementalReduce()
1864 *reduce->to_apply(), {Load(accumulator_addr), input_element}, in EmitElementalReduce()
1866 Store(result, accumulator_addr); in EmitElementalReduce()
1869 return Load(accumulator_addr); in EmitElementalReduce()