Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc742 llvm::Value* in_bounds_condition = b_.getTrue(); in HandleSelectAndScatter() local
752 in_bounds_condition = And(in_bounds_condition, index_condition); in HandleSelectAndScatter()
754 CHECK(in_bounds_condition != nullptr); in HandleSelectAndScatter()
759 llvm_ir::EmitIfThenElse(in_bounds_condition, "in-bounds", &b_); in HandleSelectAndScatter()
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc2901 llvm::Value* in_bounds_condition = b_->getInt1(true); in EmitConvolution() local
2910 in_bounds_condition = And(in_bounds_condition, dim_ok); in EmitConvolution()
2924 llvm_ir::EmitIfThenElse(in_bounds_condition, "in-bounds", b_); in EmitConvolution()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc2305 llvm::Value* in_bounds_condition = b_.getInt1(true); in EmitSelectAndScatterFromMlir() local
2323 in_bounds_condition = And(in_bounds_condition, index_condition); in EmitSelectAndScatterFromMlir()
2329 llvm_ir::EmitIfThenElse(in_bounds_condition, "in-bounds", &b_); in EmitSelectAndScatterFromMlir()