Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dparallel_loop_emitter.cc142 auto if_in_bounds = llvm_ir::EmitIfThenElse( in EmitIndexAndSetExitBasicBlock() local
149 exit_bb_ = if_in_bounds.after_block; in EmitIndexAndSetExitBasicBlock()
153 llvm_ir::SetToFirstInsertPoint(if_in_bounds.true_block, b_); in EmitIndexAndSetExitBasicBlock()
Dir_emitter_unnested.cc2328 llvm_ir::LlvmIfData if_in_bounds = in EmitSelectAndScatterFromMlir() local
2330 llvm_ir::SetToFirstInsertPoint(if_in_bounds.true_block, &b_); in EmitSelectAndScatterFromMlir()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlhlo_legalize_to_parallel_loops.cc616 auto if_in_bounds = inner_loop_b.create<scf::IfOp>( in SelectIvs() local
623 if_in_bounds.getThenBodyBuilder(b->getListener()); in SelectIvs()
632 if_in_bounds.getElseBodyBuilder(b->getListener()); in SelectIvs()
636 inner_loop_b.create<scf::YieldOp>(loc, if_in_bounds.getResults()); in SelectIvs()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc758 llvm_ir::LlvmIfData if_in_bounds = in HandleSelectAndScatter() local
760 SetToFirstInsertPoint(if_in_bounds.true_block, &b_); in HandleSelectAndScatter()