Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlhlo_legalize_to_parallel_loops.cc605 WindowLoops window_loops = InsertWindowLoops(s_and_s_op, loop_over_src, b); in SelectIvs() local
607 OpBuilder::atBlockEnd(window_loops.inner_loop.getBody()); in SelectIvs()
612 window_loops.window_ivs, &inner_loop_b); in SelectIvs()
614 IterArgs ivs_val_flag(window_loops.inner_loop.getRegionIterArgs()); in SelectIvs()
617 loc, window_loops.inner_loop.getResultTypes(), mapped_ivs.in_bounds, in SelectIvs()
637 return window_loops.selected_ivs; in SelectIvs()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc729 llvm_ir::ForLoopNest window_loops(IrName(select_and_scatter, "window"), &b_); in HandleSelectAndScatter() local
734 const llvm_ir::IrArray::Index window_index = window_loops.AddLoopsForShape( in HandleSelectAndScatter()
736 SetToFirstInsertPoint(window_loops.GetInnerLoopBodyBasicBlock(), &b_); in HandleSelectAndScatter()
810 SetToFirstInsertPoint(window_loops.GetOuterLoopExitBasicBlock(), &b_); in HandleSelectAndScatter()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc2284 llvm_ir::ForLoopNest window_loops(absl::StrCat(name, "inner"), &b_, in EmitSelectAndScatterFromMlir() local
2295 const IrArray::Index window_index = window_loops.AddLoopsForShape( in EmitSelectAndScatterFromMlir()
2298 llvm_ir::SetToFirstInsertPoint(window_loops.GetInnerLoopBodyBasicBlock(), in EmitSelectAndScatterFromMlir()
2391 llvm_ir::SetToFirstInsertPoint(window_loops.GetOuterLoopExitBasicBlock(), in EmitSelectAndScatterFromMlir()