Searched refs:EmitIfThenElse (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | parallel_loop_emitter.cc | 132 auto if_in_bounds = llvm_ir::EmitIfThenElse( in EmitIndexAndSetExitBasicBlock()
|
| D | ir_emitter_unnested.cc | 884 llvm_ir::EmitIfThenElse(in_bounds_condition, "in-bounds", &b_); in HandleSelectAndScatter()
886 llvm_ir::LlvmIfData if_initialized = llvm_ir::EmitIfThenElse( in HandleSelectAndScatter()
932 llvm_ir::EmitIfThenElse(cond, "if-select-lhs", &b_); in HandleSelectAndScatter()
1170 llvm_ir::LlvmIfData if_window_in_bounds_data = llvm_ir::EmitIfThenElse( in EmitScatter()
2794 llvm_ir::LlvmIfData if_lane_id_is_zero_data = llvm_ir::EmitIfThenElse( in EmitEpilogueForReduction()
2802 llvm_ir::LlvmIfData if_output_inbound_data = llvm_ir::EmitIfThenElse( in EmitEpilogueForReduction()
|
| D | elemental_ir_emitter.cc | 424 llvm_ir::EmitIfThenElse(in_bounds, "in_bounds", b_); in MakeElementGenerator()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | kernel_support_library.cc | 63 llvm_ir::LlvmIfData if_data = llvm_ir::EmitIfThenElse(condition, name, b_); in IfWithStatus()
|
| D | llvm_util.h | 218 LlvmIfData EmitIfThenElse(llvm::Value* condition, absl::string_view name,
|
| D | llvm_util.cc | 312 LlvmIfData EmitIfThenElse(llvm::Value* condition, absl::string_view name, in EmitIfThenElse() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 711 llvm_ir::EmitIfThenElse(in_bounds_condition, "in-bounds", &b_); in EmitElementalReduceWindow()
840 llvm_ir::EmitIfThenElse(in_bounds_condition, "in-bounds", &b_); in HandleSelectAndScatter()
842 llvm_ir::LlvmIfData if_initialized = llvm_ir::EmitIfThenElse( in HandleSelectAndScatter()
882 llvm_ir::EmitIfThenElse(cond, "if-select-lhs", &b_); in HandleSelectAndScatter()
1053 llvm_ir::EmitIfThenElse(in_bounds_condition, "in-bounds", &b_); in EmitElementalConvolution()
2566 llvm_ir::EmitIfThenElse(pred_cond, "conditional", &b_); in HandleConditional()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.cc | 940 llvm_ir::LlvmIfData if_data = llvm_ir::EmitIfThenElse( in EmitErfInv()
973 llvm_ir::LlvmIfData if_data = llvm_ir::EmitIfThenElse( in EmitErfInv()
997 llvm_ir::LlvmIfData if_data_second = llvm_ir::EmitIfThenElse( in EmitErfInv()
2025 llvm_ir::EmitIfThenElse(slice_intersection, "slice_intersection", b_); in EmitElementalDynamicUpdateSlice()
2089 llvm_ir::EmitIfThenElse(in_bounds, "in_bounds", b_); in EmitElementalPad()
|