Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc1535 llvm::Align element_alignment) { in EmitInnerLoopForVectorizedReduction() argument
1555 AlignedStore(initial_value, accumulator_shard, element_alignment); in EmitInnerLoopForVectorizedReduction()
1585 AlignedLoad(accumulator[i], element_alignment); in EmitInnerLoopForVectorizedReduction()
1586 auto addend = AlignedLoad(input_address_typed, element_alignment); in EmitInnerLoopForVectorizedReduction()
1591 AlignedStore(reduced_result, accumulator[i], element_alignment); in EmitInnerLoopForVectorizedReduction()
1604 result_ssa.push_back(AlignedLoad(accumulator_shard, element_alignment)); in EmitInnerLoopForVectorizedReduction()
1669 llvm::Align element_alignment(tensorflow::MathUtil::GCD<unsigned>( in EmitVectorizedReduce()
1744 init_value, arg, dimensions, element_alignment)); in EmitVectorizedReduce()
1749 EmitShardedVectorStore(output_address, accumulator, element_alignment, in EmitVectorizedReduce()
1778 init_value, arg, dimensions, element_alignment)); in EmitVectorizedReduce()
[all …]
Dir_emitter.h402 llvm::Align element_alignment);