Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc810 llvm_ir::ForLoopNest window_loops(IrName(select_and_scatter, "window"), &b_); in HandleSelectAndScatter() local
815 const llvm_ir::IrArray::Index window_index = window_loops.AddLoopsForShape( in HandleSelectAndScatter()
817 SetToFirstInsertPoint(window_loops.GetInnerLoopBodyBasicBlock(), &b_); in HandleSelectAndScatter()
891 SetToFirstInsertPoint(window_loops.GetOuterLoopExitBasicBlock(), &b_); in HandleSelectAndScatter()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc851 llvm_ir::ForLoopNest window_loops(IrName(select_and_scatter, "inner"), &b_, in HandleSelectAndScatter() local
858 const IrArray::Index window_index = window_loops.AddLoopsForShape( in HandleSelectAndScatter()
860 llvm_ir::SetToFirstInsertPoint(window_loops.GetInnerLoopBodyBasicBlock(), in HandleSelectAndScatter()
941 llvm_ir::SetToFirstInsertPoint(window_loops.GetOuterLoopExitBasicBlock(), in HandleSelectAndScatter()