Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc1593 llvm_ir::ForLoopNest reduction_loop_nest(IrName(arg, "vectorized_inner"), in EmitInnerLoopForVectorizedReduction() local
1596 reduction_loop_nest.AddLoopsForShapeOnDimensions(arg->shape(), dimensions, in EmitInnerLoopForVectorizedReduction()
1599 SetToFirstInsertPoint(reduction_loop_nest.GetInnerLoopBodyBasicBlock(), &b_); in EmitInnerLoopForVectorizedReduction()
1634 SetToFirstInsertPoint(reduction_loop_nest.GetOuterLoopExitBasicBlock(), &b_); in EmitInnerLoopForVectorizedReduction()