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.cc1570 unsigned element_alignment) { in EmitInnerLoopForVectorizedReduction() argument
1590 AlignedStore(initial_value, accumulator_shard, element_alignment); in EmitInnerLoopForVectorizedReduction()
1620 AlignedLoad(accumulator[i], element_alignment); in EmitInnerLoopForVectorizedReduction()
1621 auto addend = AlignedLoad(input_address_typed, element_alignment); in EmitInnerLoopForVectorizedReduction()
1626 AlignedStore(reduced_result, accumulator[i], element_alignment); in EmitInnerLoopForVectorizedReduction()
1639 result_ssa.push_back(AlignedLoad(accumulator_shard, element_alignment)); in EmitInnerLoopForVectorizedReduction()
1689 unsigned element_alignment = tensorflow::MathUtil::GCD<unsigned>( in EmitVectorizedReduce() local
1764 init_value, arg, dimensions, element_alignment)); in EmitVectorizedReduce()
1769 EmitShardedVectorStore(output_address, accumulator, element_alignment, in EmitVectorizedReduce()
1798 init_value, arg, dimensions, element_alignment)); in EmitVectorizedReduce()
[all …]
Dir_emitter.h401 unsigned element_alignment);