Searched refs:element_generator (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_nested.cc | 118 const llvm_ir::ElementGenerator& element_generator) { in EmitTargetElementLoop() argument
125 llvm_ir::LoopEmitter(element_generator, target_arrays, &b_).EmitLoop()); in EmitTargetElementLoop()
129 return llvm_ir::LoopEmitter(element_generator, GetIrArray(hlo, hlo), &b_) in EmitTargetElementLoop()
|
| D | ir_emitter_unnested.cc | 2176 const llvm_ir::ElementGenerator& element_generator, KernelThunk* thunk) { in EmitTargetElementLoopInThunk() argument
2191 return ParallelLoopEmitter(element_generator, GetIrArray(hlo, hlo), in EmitTargetElementLoopInThunk()
2209 ParallelLoopEmitter(element_generator, output_arrays, launch_dimensions, in EmitTargetElementLoopInThunk()
2244 const llvm_ir::ElementGenerator& element_generator) { in EmitTargetElementLoop() argument
2255 EmitTargetElementLoopInThunk(hlo, element_generator, kernel_thunk.get()); in EmitTargetElementLoop()
3245 TileElementGenerator element_generator; in EmitHlo021Tile() local
3247 element_generator = [&](HloInstruction* hlo, in EmitHlo021Tile()
3256 element_generator = in EmitHlo021Tile()
3265 KernelCodeGenerator kernel_generator(std::move(element_generator)); in EmitHlo021Tile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | dynamic_update_slice_util.cc | 175 ElementGenerator element_generator = in EmitFusedDynamicUpdateSliceInPlaceImpl() local
177 return element_generator(IrArray::Index(b->getInt64Ty())); in EmitFusedDynamicUpdateSliceInPlaceImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.h | 312 const llvm_ir::ElementGenerator& element_generator);
315 const llvm_ir::ElementGenerator& element_generator);
|
| D | ir_emitter.cc | 2944 const llvm_ir::ElementGenerator& element_generator) { in EmitTargetElementLoop() argument
2945 return EmitTargetElementLoop(target_op, /*desc=*/"", element_generator); in EmitTargetElementLoop()
2950 const llvm_ir::ElementGenerator& element_generator) { in EmitTargetElementLoop() argument
2970 llvm_ir::LoopEmitter(element_generator, output_arrays, &b_) in EmitTargetElementLoop()
2985 TF_RETURN_IF_ERROR(ParallelLoopEmitter(element_generator, target_array, in EmitTargetElementLoop()
2990 llvm_ir::LoopEmitter(element_generator, target_array, &b_) in EmitTargetElementLoop()
|