Searched refs:window_loops (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
D | lhlo_legalize_to_parallel_loops.cc | 605 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/ |
D | ir_emitter.cc | 729 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/ |
D | ir_emitter_unnested.cc | 2284 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()
|