Searched refs:EmitIfThenElse (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | parallel_loop_emitter.cc | 142 auto if_in_bounds = llvm_ir::EmitIfThenElse( in EmitIndexAndSetExitBasicBlock()
|
| D | ir_emitter_unnested.cc | 910 auto if_in_dyn_bounds = llvm_ir::EmitIfThenElse( in EmitPadToStaticFromMlir()
1031 auto if_in_dyn_bounds = llvm_ir::EmitIfThenElse( in EmitSliceToDynamicFromMlir()
2329 llvm_ir::EmitIfThenElse(in_bounds_condition, "in-bounds", &b_); in EmitSelectAndScatterFromMlir()
2331 llvm_ir::LlvmIfData if_initialized = llvm_ir::EmitIfThenElse( in EmitSelectAndScatterFromMlir()
2382 llvm_ir::EmitIfThenElse(cond, "if-select-lhs", &b_); in EmitSelectAndScatterFromMlir()
2692 llvm_ir::LlvmIfData if_window_in_bounds_data = llvm_ir::EmitIfThenElse( in EmitScatter()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | kernel_support_library.cc | 64 llvm_ir::EmitIfThenElse(condition, name, b_, in IfWithStatus()
|
| D | llvm_util.h | 213 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 | 759 llvm_ir::EmitIfThenElse(in_bounds_condition, "in-bounds", &b_); in HandleSelectAndScatter()
761 llvm_ir::LlvmIfData if_initialized = llvm_ir::EmitIfThenElse( in HandleSelectAndScatter()
801 llvm_ir::EmitIfThenElse(cond, "if-select-lhs", &b_); in HandleSelectAndScatter()
2668 llvm_ir::EmitIfThenElse(pred_cond, "conditional", &b_); in HandleConditional()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.cc | 2084 llvm_ir::EmitIfThenElse(slice_intersection, "slice_intersection", b_); in EmitElementalDynamicUpdateSlice()
2148 llvm_ir::EmitIfThenElse(in_bounds, "in_bounds", b_); in EmitElementalPad()
2681 llvm_ir::EmitIfThenElse(in_bounds, "in_bounds", b_); in EmitElementalReduceWindow()
2924 llvm_ir::EmitIfThenElse(in_bounds_condition, "in-bounds", b_); in EmitConvolution()
|