Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc669 llvm::Value* in_bounds_condition = nullptr; in EmitElementalReduceWindow() local
684 if (in_bounds_condition == nullptr) { in EmitElementalReduceWindow()
685 in_bounds_condition = dilation_condition; in EmitElementalReduceWindow()
687 in_bounds_condition = And(in_bounds_condition, dilation_condition); in EmitElementalReduceWindow()
702 if (in_bounds_condition == nullptr) { in EmitElementalReduceWindow()
703 in_bounds_condition = index_condition; in EmitElementalReduceWindow()
705 in_bounds_condition = And(in_bounds_condition, index_condition); in EmitElementalReduceWindow()
708 CHECK(in_bounds_condition != nullptr); in EmitElementalReduceWindow()
711 llvm_ir::EmitIfThenElse(in_bounds_condition, "in-bounds", &b_); in EmitElementalReduceWindow()
823 llvm::Value* in_bounds_condition = b_.getTrue(); in HandleSelectAndScatter() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc867 llvm::Value* in_bounds_condition = b_.getInt1(true); in HandleSelectAndScatter() local
877 in_bounds_condition = And(in_bounds_condition, index_condition); in HandleSelectAndScatter()
879 CHECK(in_bounds_condition != nullptr); in HandleSelectAndScatter()
884 llvm_ir::EmitIfThenElse(in_bounds_condition, "in-bounds", &b_); in HandleSelectAndScatter()